Use PyTorch Deep Learning Models with scikit-learn

The most popular deep learning libraries in Python for research and development are TensorFlow/Keras and PyTorch, due to their simplicity. The scikit-learn library, however, is the most popular library for general machine learning in Python. In this post, you will discover how to use deep learning models from PyTorch with the scikit-learn library in Python. This will allow you to leverage the power of the scikit-learn library for tasks like model evaluation and model hyper-parameter optimization. After completing this lesson you will know:

• How to wrap a PyTorch model for use with the scikit-learn machine learning library
• How to easily evaluate PyTorch models using cross-validation in scikit-learn
• How to tune PyTorch model hyperparameters using grid search in scikit-learn

Kick-start