Neptune.ai scikit-learn integration library
Project description
Neptune + scikit-learn integration
Experiment tracking for scikit-learn–trained models.
What will you get with this integration?
- Log, organize, visualize, and compare ML experiments in a single place
- Monitor model training live
- Version and query production-ready models and associated metadata (e.g., datasets)
- Collaborate with the team and across the organization
What will be logged to Neptune?
- classifier and regressor parameters,
- pickled model,
- test predictions,
- test predictions probabilities,
- test scores,
- classifier and regressor visualizations, like confusion matrix, precision-recall chart, and feature importance chart,
- KMeans cluster labels and clustering visualizations,
- metadata including git summary info,
- other metadata
Resources
Example
# On the command line:
pip install neptune-sklearn
# In Python, prepare a fitted estimator
parameters = {
"n_estimators": 70, "max_depth": 7, "min_samples_split": 3
}
estimator = ...
estimator.fit(X_train, y_train)
# Import Neptune and start a run
import neptune
run = neptune.init_run(
project="common/sklearn-integration",
api_token=neptune.ANONYMOUS_API_TOKEN,
)
# Log parameters and scores
run["parameters"] = parameters
y_pred = estimator.predict(X_test)
run["scores/max_error"] = max_error(y_test, y_pred)
run["scores/mean_absolute_error"] = mean_absolute_error(y_test, y_pred)
run["scores/r2_score"] = r2_score(y_test, y_pred)
# Stop the run
run.stop()
Support
If you got stuck or simply want to talk to us, here are your options:
- Check our FAQ page
- You can submit bug reports, feature requests, or contributions directly to the repository.
- 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),
- You can just shoot us an email at support@neptune.ai
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
neptune_sklearn-2.1.4.tar.gz
(14.6 kB
view details)
Built Distribution
File details
Details for the file neptune_sklearn-2.1.4.tar.gz
.
File metadata
- Download URL: neptune_sklearn-2.1.4.tar.gz
- Upload date:
- Size: 14.6 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.9.19
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 9687a4cc8349e5378766c48fe014ccaf955e155c40f9cc2fd38ebf0fd50a4fd7 |
|
MD5 | 64802a7a768f856eab0e3fbd1ce7d619 |
|
BLAKE2b-256 | 9ba5a7f5aa08f481ef03acd1cc296536a4f0cf1c14a8508c992d7dbb597665a7 |
File details
Details for the file neptune_sklearn-2.1.4-py3-none-any.whl
.
File metadata
- Download URL: neptune_sklearn-2.1.4-py3-none-any.whl
- Upload date:
- Size: 15.0 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.9.19
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | f0fce972ecf38e801a51aa611b226d00f9a16f0b192c0382b1751d3bccb0ff3f |
|
MD5 | d9c8f41a9a619aa01097700210eeb006 |
|
BLAKE2b-256 | 8c63eb829b258168b99d5932353102be0be245a27ead2995c60933cc0b1d171e |