A set of functions and demos to make machine learning projects easier to understand through effective visualizations.
Project description
MLVisualizationTools
MLVisualizationTools is a python library to make machine learning more understandable through the use of effective visualizations.
We support graphing with matplotlib and plotly. We implicity support all major ML libraries, such as tensorflow and sklearn.
You can use the built in apps to quickly anaylyze your existing models, or build custom projects using the modular sets of functions.
Installation
pip install MLVisualizationTools
Depending on your use case, tensorflow, plotly and matplotlib might need to be installed.
pip install tensorflow
pip install plotly
pip install matplotlib
To use interactive webapps, use the pip install MLVisualizationTools[dash]
or pip install MLVisualizationTools[dash-notebook]
flags on install.
If you are running on a notebook that doesn't have dash support (like kaggle), you might need
pip install MLVisualizationTools[ngrok-tunneling]
Express
To get started using MLVisualizationTools, run one of the prebuilt apps.
import MLVisualizationTools.express.DashModelVisualizer as App
model = ... #your keras model
data = ... #your pandas dataframe with features
App.visualize(model, data)
Functions
MLVisualizationTools connects a variety of smaller functions.
Steps:
- Keras Model and Dataframe with features
- Analyzer
- Interface / Interface Raw (if you don't have a dataframe)
- Colorizers (optional)
- Graphs
Analyzers take a keras model and return information about the inputs such as which ones have high variance.
Interfaces take parameters and construct a multidimensional grid of values based on plugging these numbers into the model.
(Raw interfaces allow you to use interfaces by specifying column data instead of a pandas dataframe. Column data is a list with a dict with name, min, max, and mean values for each feature column)
Colorizers mark points as being certain colors, typically above or below 0.5.
Graphs turn these output grids into a visual representation.
Sample
from MLVisualizationTools import Analytics, Interfaces, Graphs, Colorizers
#Displays plotly graphs with max variance inputs to model
model = ... #your model
df = ... #your dataframe
AR = Analytics.analyzeModel(model, df)
maxvar = AR.maxVariance()
grid = Interfaces.predictionGrid(model, maxvar[0].name, maxvar[1].name, df)
grid = Colorizers.binary(grid)
fig = Graphs.plotlyGraph(grid)
fig.show()
Prebuilt Examples
Prebuilt examples run off of the pretrained model and dataset packaged with this library. They include:
- Demo: a basic demo of library functionality that renders 2 plots
- MatplotlibDemo: Demo but with matplotlib instead of plotly
- DashDemo: Non-jupyter notebook version of an interactive dash website demo
- DashNotebookDemo: Notebook version of an interactive website demo
- DashKaggleDemo: Notebook version of an dash demo that works in kaggle notebooks
See MLVisualizationTools/Examples for more examples. Use example.main() to run the examples and set parameters such as themes.
Support for more ML Libraries
We support any ML library that has a predict()
call that takes
a pd Dataframe with features. If this doesn't work, use a wrapper class like
in this example:
import pandas as pd
class ModelWrapper:
def __init(self, model):
self.model = model
def predict(self, dataframe: pd.DataFrame):
... #Do whatever code you need here
Tensorflow Compatibility
MLVisualizationTools is distributed with a pretrained tensorflow model to make running examples quick and easy. It is not needed for main library functions.
For version 2.0 through 2.4, we load a v2.0 model. For version 2.5+ we load a v2.5 model.
If this causes compatibility issues you can still use the main library on your models. If you need an example model, retrain it with TrainTitanicModel.py
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
Built Distribution
Hashes for MLVisualizationTools-0.4.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | e42595b42490a24599c6ebe6e70144ac4093a7dfc1802f5232af39788812612a |
|
MD5 | e8666b96744c748b05b3f0ba5f0ac539 |
|
BLAKE2b-256 | 7b365b1ba1a11a64ef6795ced3fc481a12d13e0a86809d84aaa6a6527b0af5cb |
Hashes for MLVisualizationTools-0.4.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3a5b10aee5e070de8b9463fb04d9adcab96bfe0804cd461fd2cb2531dd05060d |
|
MD5 | 2b6226480b89ff3817aa185d8f59b44c |
|
BLAKE2b-256 | 2aa7cbd524af33164fae42535c7058dc7fd0281bb1545e2181a51330b7572458 |