Colloquium 2023 Skorch: Training PyTorch models with scikit-learn

From SHARCNETHelp
Jump to navigationJump to search

PyTorch is an enormously popular framework for developing deep learning models in Python. However, scikit-learn is one of the most popular libraries for general machine learning. Skorch is a wrapper for PyTorch that allows one to use models written with PyTorch with the scikit-learn library. In this talk, we will explore how skorch allows for PyTorch models to be easily incorporated into scikit-learn data/training pipelines, cross-validation and hyperparameter search schemes and eliminates the need for boilerplate training code.