Scikit Learn Pytorch