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

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.