Please welcome skorch 1.1.0 - a smaller release with a few fixes, a new notebook showcasing learning rate
schedulers and mainly support for scikit-learn 1.6.0.
Full list of changes:
Added
Changed
- All neural net classes now inherit from sklearn's
BaseEstimator
. This is to support compatibility with sklearn 1.6.0 and above. Classification models additionally inherit fromClassifierMixin
and regressors fromRegressorMixin
. (#1078) - When using the
ReduceLROnPlateau
learning rate scheduler, we now record the learning rate in the net history (net.history[:, 'event_lr']
by default). It is now also possible to to step per batch, not only by epoch (#1075) - The learning rate scheduler
.simulate()
method now supports adding step args which is useful when simulation policies such asReduceLROnPlateau
which expect metrics to base their schedule on. (#1077) - Removed deprecated
skorch.callbacks.scoring.cache_net_infer
(#1088)
Fixed
- Fix an issue with using
NeuralNetBinaryClassifier
withtorch.compile
(#1058)
Thanks @Ball-Man and @ParagEkbote for their contributions.