Scikit learn pytorch