Pytorch Scikit Learn