openml / automlbenchmark

OpenML AutoML Benchmarking Framework
https://openml.github.io/automlbenchmark
MIT License
399 stars 132 forks source link

Add `optimize_for_deployment` for AutoGluon_hq #544

Closed Innixma closed 1 year ago

Innixma commented 1 year ago

Add optimize_for_deployment for AutoGluon_hq

This disables some of the post-fit logic in favor of a more optimized fit process that reduces disk usage by >10x and reduces overall fit time without any compromise to result quality or inference speed.

Not necessarily critical for the upcoming benchmark run, but it will lead to slightly faster training times due to only the necessary models being refit, rather than all models being refit, making it less likely that AutoGluon goes significantly above the time limit.