Resources

How to Track ML Model Training: LightGBM + neptune.ai Integration

1 min
Prince Canuma
4th April, 2022

What will you get with this integration?

LightGBM is a gradient boosting framework that uses tree-based learning algorithms.

Neptune + LightGBM integration, lets you:

  • Automatically log many types of metadata during trainingrn
    • Training and validation metrics

    • Parameters

    • Feature names, num_features and num_rows for the train set

    • Hardware consumption (CPU, GPU, memory)

    • Stdout and stderr logs

    • Training code and git commit information

  • Log model summary after trainingrn
    • Pickled model

    • Feature importance chart (gain and split)

    • Visualized trees

    • Trees saved as DataFrame

    • Confusion matrix (only for classification problems)

Read the documentation

Couldn’t find the use case you were looking for?

Just get in touch, and our ML team will create a custom demo for you.

    Contact with us

    This site is protected by reCAPTCHA and the Google Privacy Policy and Terms of Service apply.

    * - required fields