Nixtla / mlforecast

Scalable machine 🤖 learning for time series forecasting.
https://nixtlaverse.nixtla.io/mlforecast
Apache License 2.0
789 stars 74 forks source link

[MLForecast: test set eval and early stopping ] #309

Closed Djuseppe closed 5 months ago

Djuseppe commented 5 months ago

Description

Hey team, it's gonna be beneficial to have an option to early stop models when training.

Use case

CV hyper-param optimization

jmoralez commented 5 months ago

Hey @Djuseppe, thanks for using mlforecast. We have that built-in for LightGBM in the LightGBMCV class and you can do it for other frameworks by using custom training. It'd be very hard to have a general solution because each framework has a slightly different interface.

Djuseppe commented 5 months ago

thanks for the info 🙂 indeed a custom training is amazing and suits me nicely, since I have also custom CV with Optuna optimization. Closing, as existing features are just enough.