{"id":14970687,"url":"https://github.com/neptune-ai/neptune-sklearn","last_synced_at":"2025-07-20T02:32:09.113Z","repository":{"id":37700592,"uuid":"362730458","full_name":"neptune-ai/neptune-sklearn","owner":"neptune-ai","description":"Experiment tracking for scikit-learn. 🧩 Log, organize, visualize and compare model metrics, parameters, dataset versions, and more.","archived":false,"fork":false,"pushed_at":"2024-07-05T13:47:15.000Z","size":117,"stargazers_count":6,"open_issues_count":0,"forks_count":2,"subscribers_count":10,"default_branch":"master","last_synced_at":"2025-07-18T15:16:35.103Z","etag":null,"topics":["dashboard","logger","management","metadata","mlops","optimization","organize","scikit-learn","sklearn","store","tracker","training","versioning"],"latest_commit_sha":null,"homepage":"https://docs.neptune.ai/integrations/sklearn/","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"apache-2.0","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/neptune-ai.png","metadata":{"files":{"readme":"README.md","changelog":"CHANGELOG.md","contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null}},"created_at":"2021-04-29T07:36:36.000Z","updated_at":"2024-07-05T13:44:12.000Z","dependencies_parsed_at":"2022-08-24T17:10:16.930Z","dependency_job_id":"969ab334-56e7-45ab-9165-b2a8ee634473","html_url":"https://github.com/neptune-ai/neptune-sklearn","commit_stats":{"total_commits":65,"total_committers":11,"mean_commits":5.909090909090909,"dds":0.676923076923077,"last_synced_commit":"0f201db8316712b4dc9c0d6e46754e7c6ecb851b"},"previous_names":[],"tags_count":12,"template":false,"template_full_name":null,"purl":"pkg:github/neptune-ai/neptune-sklearn","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/neptune-ai%2Fneptune-sklearn","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/neptune-ai%2Fneptune-sklearn/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/neptune-ai%2Fneptune-sklearn/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/neptune-ai%2Fneptune-sklearn/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/neptune-ai","download_url":"https://codeload.github.com/neptune-ai/neptune-sklearn/tar.gz/refs/heads/master","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/neptune-ai%2Fneptune-sklearn/sbom","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":265937374,"owners_count":23852112,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["dashboard","logger","management","metadata","mlops","optimization","organize","scikit-learn","sklearn","store","tracker","training","versioning"],"created_at":"2024-09-24T13:43:59.501Z","updated_at":"2025-07-20T02:32:09.064Z","avatar_url":"https://github.com/neptune-ai.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"# Neptune + scikit-learn integration\n\nExperiment tracking for scikit-learn\u0026ndash;trained models.\n\n## What will you get with this integration?\n\n* Log, organize, visualize, and compare ML experiments in a single place\n* Monitor model training live\n* Version and query production-ready models and associated metadata (e.g., datasets)\n* Collaborate with the team and across the organization\n\n## What will be logged to Neptune?\n\n* classifier and regressor parameters,\n* pickled model,\n* test predictions,\n* test predictions probabilities,\n* test scores,\n* classifier and regressor visualizations, like confusion matrix, precision-recall chart, and feature importance chart,\n* KMeans cluster labels and clustering visualizations,\n* metadata including git summary info,\n* [other metadata](https://docs.neptune.ai/logging/what_you_can_log)\n\n![image](https://docs.neptune.ai/img/app/integrations/scikit-learn.png)\n\n## Resources\n\n* [Documentation](https://docs.neptune.ai/integrations/sklearn)\n* [Code example on GitHub](https://github.com/neptune-ai/examples/blob/main/integrations-and-supported-tools/sklearn/scripts/Neptune_Scikit_learn_classification.py)\n* [Runs logged in the Neptune app](https://app.neptune.ai/o/common/org/sklearn-integration/e/SKLEAR-95/all)\n* [Run example in Google Colab](https://colab.research.google.com/github/neptune-ai/examples/blob/master/integrations-and-supported-tools/sklearn/notebooks/Neptune_Scikit_learn.ipynb)\n\n## Example\n\n```\n# On the command line:\npip install neptune-sklearn\n```\n\n```python\n# In Python, prepare a fitted estimator\nparameters = {\n    \"n_estimators\": 70, \"max_depth\": 7, \"min_samples_split\": 3\n}\n\nestimator = ...\nestimator.fit(X_train, y_train)\n\n# Import Neptune and start a run\nimport neptune\n\nrun = neptune.init_run(\n    project=\"common/sklearn-integration\",\n    api_token=neptune.ANONYMOUS_API_TOKEN,\n)\n\n# Log parameters and scores\nrun[\"parameters\"] = parameters\n\ny_pred = estimator.predict(X_test)\n\nrun[\"scores/max_error\"] = max_error(y_test, y_pred)\nrun[\"scores/mean_absolute_error\"] = mean_absolute_error(y_test, y_pred)\nrun[\"scores/r2_score\"] = r2_score(y_test, y_pred)\n\n# Stop the run\nrun.stop()\n```\n\n## Support\n\nIf you got stuck or simply want to talk to us, here are your options:\n\n* Check our [FAQ page](https://docs.neptune.ai/getting_help)\n* You can submit bug reports, feature requests, or contributions directly to the repository.\n* Chat! When in the Neptune application click on the blue message icon in the bottom-right corner and send a message. A real person will talk to you ASAP (typically very ASAP),\n* You can just shoot us an email at support@neptune.ai\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fneptune-ai%2Fneptune-sklearn","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fneptune-ai%2Fneptune-sklearn","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fneptune-ai%2Fneptune-sklearn/lists"}