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.
Find the code for my tips here: GitHub.
I like to explore, experiment and write about data science concepts and tools. You can read my articles on Medium. Also, you can connect with me on LinkedIn.