search

microsoft / FLAML

A fast library for AutoML and tuning. Join our Discord: https://discord.gg/Cppx2vSPVP.
https://microsoft.github.io/FLAML/
MIT License
3.91k stars 508 forks source link

Feature request: Allow for optional inclusion of catboost, lgbm and xgboost #252

Closed camer314 closed 3 years ago

camer314 commented 3 years ago

These packages incur massive additional cost to size required by FLAML when used in a docker image (seems like installing FLAML adds ~1Gb to image).

It would be awesome if it was possible to remove dependencies on libraries which will not be used at runtime. For example, we are unlikely to be using xgboost or catboost or possible not even using LGBM as we have mostly our own estimators so would like to uninstall these but FLAML complains about missing packages

sonichi commented 3 years ago

Dependency on catboost is already removed. We can do the same for lgbm and xgboost.