Skorch: Use Scikit-learn API on PyTorch Models
skorch is a high-level library for PyTorch that provides full Scikit-learn compatibility. In other words, it combines the power of PyTorch with the elegance of sklearn.
Thus, you can train PyTorch models in a way similar to Scikit-learn, using functions such as fit, predict, score, etc.
Using skorch, you can also put a PyTorch model in the sklearn pipeline, and many more.
Overall, it aims at being as flexible as PyTorch while having a clean interface as sklearn.
Read more: Documentation.
Share this post on LinkedIn: Post Link.
Thanks for reading Daily Dose of Data Science! Subscribe for free to learn something new and insightful about Python and Data Science every day. Also, get a Free Data Science PDF (250+ pages) with 200+ tips.
Find the code for my tips here: GitHub.