Colloquium 2023 Skorch: Training PyTorch models with scikit-learn

From SHARCNETHelp
Revision as of 10:48, 27 November 2023 by Syam (talk | contribs) (Created page with "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 learni...")
(diff) ← Older revision | Latest revision (diff) | Newer revision → (diff)
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.