search

Neuraxio / Neuraxle

The world's cleanest AutoML library ✨ - Do hyperparameter tuning with the right pipeline abstractions to write clean deep learning production pipelines. Let your pipeline steps have hyperparameter spaces. Design steps in your pipeline like components. Compatible with Scikit-Learn, TensorFlow, and most other libraries, frameworks and MLOps environments.
https://www.neuraxle.org/
Apache License 2.0
608 stars 62 forks source link

Bug: StepSaverCallback & BestModelCheckpoint not working in version 0.7.0 #522

Closed guillaume-chevalier closed 9 months ago

guillaume-chevalier commented 2 years ago

Describe the bug From version 0.7.0 and later, these callbacks don't have access to the pipeline anymore since the pipeline isn't stored anymore in the TrialSplit, but just in the AutoML class.

To Reproduce Update to 0.7.0 and run these tests that are skipped.

Expected behavior Callbacks are supposed to log models using the repo.

Suggested Fix Possibly have the TrialSplits contain their trained model again, instead of letting the Trainer call the model.

Additional context 0.7.0 is not merged nor deployed at the moment of writing this issue.

stale[bot] commented 1 year ago

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs in the next 180 days. Thank you for your contributions.