Pytorch vs sklearn.