Contribute Media
A thank you to everyone who makes this possible: Read More

Skorch: A Union of Scikit learn and PyTorch

Description

Skorch is a scikit-learn compatible neural network library that wraps PyTorch. skorch reduces the boilerplate needed to train a neural network by abstracting away the training loop and providing a callback API for common tasks such as recording metrics and model checkpointing. This talk is targeted to users familiar with the scikit-learn API and have had some exposure to neural networks.

Details

Improve this page