Error while training LightGBM model
I have installed LightGBM plugin but while training a model I get error `Training Failed… Sorry`
Failed to train : <type 'exceptions.ImportError'> : No module named lightgbm
Can someone help me to overcome this error please.
Best Answer
-
Hi,
Can you please make sure you are using a code environment for this training where the "lightGBM" package is installed? More details can be found in the Installation section of the plugin documentation here: https://www.dataiku.com/product/plugins/model-lightgbm/
Thanks,
Andrew
Answers
-
yash Partner, L2 Designer, Dataiku DSS Core Designer, Dataiku DSS ML Practitioner, Dataiku DSS Core Concepts, Registered Posts: 7 Partner
Thank you that helped solve the issue.
However I have a doubt, do I need to create a code env just because it is installed through plugin? Because I could run other algorithms XGBoost, GBM etc successfully without any issues when I didn't create a code env.
-
Hi,
That's correct. All algorithms that come builtin with DSS are already present and don't need to be installed in a separate environment. LightGBM isn't, hence the need for a code environment.
-
yash Partner, L2 Designer, Dataiku DSS Core Designer, Dataiku DSS ML Practitioner, Dataiku DSS Core Concepts, Registered Posts: 7 Partner
Thanks for your response.