Machine learning with PyTorch and scikit-learn