Convert scikit-learn models to PyTorch modules - View it on GitHub