Skorch is a scikit-learn
compatible neural network library that wraps PyTorch. skorch reduces the boilerplate needed to train a neutral network by abstracting away the training loop and providing a callback API for common tasks such as recording metrics. This talk is targeted to users familiar with the scikit-learn API and have had some exposure to neutral networks.
Skorch is a scikit-learn
compatible neural network library that wraps PyTorch. Training a neutral network in pure PyTorch requires writing: a training loop, metrics recording, model checkpointing, and other boilerplate code. skorch remedies this issue by abstracting away the training loop and providing a callback API for common tasks. This talk is targeted to users familiar with the scikit-learn API and have had some exposure to neutral networks. First, I will motivate the use-case for skorch and introduce the API. Next, I will go through concrete problems and how to use skorch's features to solve them. By the end of the talk, the audience will learn how to train a neural network using skorch, and how to extend skorch to fit their needs.
scikit-learn
's GridSearchCV
and Pipeline
.Dataset
object with skorch.