Na verdade nem isso eu sei. Trabalho com golang e não achei biblioteca boa para ML, então escrevi uma api em python que usa scikit e toda a parte de python foi escrita conversando com o Gepeto (ChatGPT). Dessa forma eu deixo um serviço no docker rodando o a modelagem que preciso em python e o resto continua em Go.