SKORCH Files

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

• Added a notebook that shows how to use Learning Rate Scheduler in skorch.(#1074)

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 from ClassifierMixin and regressors from RegressorMixin. (#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 as ReduceLROnPlateau 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 with torch.compile (#1058)

Thanks @Ball-Man and @ParagEkbote for their contributions.