Fancy data functions that will make your life as a data scientist easier.
Project description
WhiteBox Utilities Toolkit: Tools to make your life easier
Fancy data functions that will make your life as a data scientist easier.
Installing
To install this library in your Python environment:
pip install whiteboxml
Documentation
Metrics
Classification
- ROC curve / AUC:
import numpy as np
from whiteboxml.modeling.metrics import plot_roc_auc_binary
y_pred = np.random.normal(0, 1, 1000)
y_true = np.random.choice([0, 1], 1000)
ax, fpr, tpr, thr, auc_score = plot_roc_auc_binary(y_pred=y_pred, y_true=y_true, figsize=(8, 8))
ax.get_figure().savefig('roc_curve.png')
- Confusion Matrix:
import numpy as np
from whiteboxml.modeling.metrics import plot_confusion_matrix
y_true = np.random.choice([0, 1, 2, 3], 10000)
y_pred = np.random.choice([0, 1, 2, 3], 10000)
ax, matrix = plot_confusion_matrix(y_pred=y_pred, y_true=y_true,
class_labels=['a', 'b', 'c', 'd'])
ax.get_figure().savefig('confusion_matrix.png')
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
whiteboxml-0.0.1.tar.gz
(15.4 kB
view hashes)
Built Distribution
whiteboxml-0.0.1-py3-none-any.whl
(15.8 kB
view hashes)
Close
Hashes for whiteboxml-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e53d8dd6d6820e46142a954e6e8c8cfa8faa84882d13b101121b0dd7e800e232 |
|
MD5 | ae2c4abb68dcdac35a22d3861e3d7a0d |
|
BLAKE2b-256 | a3738d296a9e8381f128f97172ffebfad3ce8f7e201429df3de9e6cd1576d5c5 |