We Raised $8M Series A to Continue Building Experiment Tracking and Model Registry That “Just Works”

Read more

Resources » How to Track ML Model Training: Scikit-learn + Neptune Integration

How to Track ML Model Training: Scikit-learn + Neptune Integration

What will you get with this integration?

Scikit-learn is an open-source machine learning framework commonly used for building predictive models. Neptune helps with keeping track of model training metadata.

With Neptune + Sklearn integration you can track your classifiers, regressors, and k-means clustering results, specifically:

  • Log classifier and regressor parameters
  • Log pickled model
  • Log test predictions
  • Log test predictions probabilities
  • Log test scores
  • Log classifier and regressor visualizations, like confusion matrix, precision-recall chart, and feature importance chart
  • Log KMeans cluster labels and clustering visualizations
  • Log metadata including git summary info

Check 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