← all repositories

skorch-dev/skorch

A library that wraps PyTorch neural networks with a scikit-learn compatible API for easier training and integration.

6.2k stars Jupyter Notebook ML Frameworks
skorch
Velocity · 7d
+1.9
★ / day
Trend
steady
star history

Skorch provides a scikit-learn compatible interface for PyTorch neural networks, allowing users to leverage PyTorch’s deep learning capabilities within scikit-learn workflows. It wraps PyTorch modules as scikit-learn estimators, enabling use with pipelines, cross-validation, and grid search. The library aims to combine the flexibility of PyTorch with the familiar interface of scikit-learn.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.