Resources

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

1 min
Prince Canuma
4th April, 2022

What will you get with this integration?

XGBoost is an optimized distributed gradient boosting library designed to be highly efficient, flexible, and portable. It implements machine learning algorithms under the Gradient Boosting framework.

Neptune + XGBoost integration, lets you automatically log many types of metadata during training:

  • Metrics
  • Parameters
  • Learning rate
  • Pickled model
  • Visualizations (feature importance chart and tree visualizations)
  • Hardware consumption (CPU, GPU, memory)
  • Stdout and stderr logs
  • Training code
  • Git commit information

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