Python Pytorch Scikit Learn