Skip to content

Version 1.1.0

Latest
Compare
Choose a tag to compare
@githubnemo githubnemo released this 10 Jan 13:04
· 2 commits to master since this release
6008085

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.