Check out the first Dataiku 8 Deep Dive focusing on Productivity on October 29th Read More

Error while training LightGBM model

Level 2
Level 2
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

error.PNG

Can someone help me to overcome this error please.

 

0 Kudos
4 Replies
Dataiker
Dataiker

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

0 Kudos
Level 2
Level 2
Author

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.

0 Kudos
Dataiker
Dataiker

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.

0 Kudos
Level 2
Level 2
Author

Thanks for your response.

0 Kudos