How to Track ML Model Training: Scikit-learn + Neptune Integration
- mins read
- Updated July 5th, 2022
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