VisualML: Visualization of Multi-Dimensional Machine Learning Models
Project description
Visual ML is a library for visualizing the decision boundary of
machine learning models from Sklearn using 2D projections of pairs
of features. Here's an example:
```
>>> import visualml as vml
>>> import pandas as pd
>>> from sklearn.datasets import make_classification
>>> from sklearn.ensemble import RandomForestClassifier as RF
>>> # Create a toy classification dataset
>>> feature_names = ['A','B','C','D']
>>> X, y = make_classification(n_features=4, random_state=42)
>>> # The visualization is only supported if X is a pandas df
>>> X = pd.DataFrame(X, columns=feature_names)
>>> # Train a classifier
>>> clf = RF(random_state=42).fit(X,y)
>>> # Plot decision boundary grid
>>> vml.decision_boundary_grid(clf, X, y)
```
machine learning models from Sklearn using 2D projections of pairs
of features. Here's an example:
```
>>> import visualml as vml
>>> import pandas as pd
>>> from sklearn.datasets import make_classification
>>> from sklearn.ensemble import RandomForestClassifier as RF
>>> # Create a toy classification dataset
>>> feature_names = ['A','B','C','D']
>>> X, y = make_classification(n_features=4, random_state=42)
>>> # The visualization is only supported if X is a pandas df
>>> X = pd.DataFrame(X, columns=feature_names)
>>> # Train a classifier
>>> clf = RF(random_state=42).fit(X,y)
>>> # Plot decision boundary grid
>>> vml.decision_boundary_grid(clf, X, y)
```
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
visualml-0.1b7.tar.gz
(12.2 kB
view hashes)