A deep learning model visualization suite based on Mermaid
Project description
Model Visual
⭐ Star us on GitHub — it motivates us a lot!
Model Visual is a lightweight deep learning model visualization suite based on Mermaid.This kit will convert model schemas into Mermaid code and output code snippets or complete web pages. Users can easily adjust the color and border of the chart.
Only support tensorflow now. More frameworks will comming soon!
Contents
Demo
This is the suite's demo. Suite will generate a web page that include a chart.
Tutorial
Isnstallation
First,install the suite by using pip.
pip install model-visual
Create object
Then,creat an object and set some parameter.
Test is the name of an object, you can take any name you want. And model is your keras model's name.(Notice: the model has to be compiled)
from model_visual import ModelVisual
test = ModelVisual(model)
Run and get result
test.run()
test.save_web_page()
Beside save_web_page()
,you can also use return_js_code()
or return_web_page()
.
save_web_page()
will create a html file,
return_js_code()
and save_web_page()
will return you sourse code.
APIs
set_name(): # set the html file name
set_path(): # set the html path name
set_chart_fill_color(): # set the cart fill color
set_chart_stroke_color(): # set the chart strole color
set_chart_stroke_width(): # set the chart stroke width
set_model(): # set the keras model
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 Model_Visual-1.1.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 65d6abd8cdf9888ec47aef39f455ad8c0cc967a5893b8b6b9c45cbc8c43dd8f8 |
|
MD5 | 360a4d5abc7339465fe19cb78b967065 |
|
BLAKE2b-256 | 90e26b8c44e886d79c8e92bc4c11ff78a59c2c8236f1d9e0dca29f253a969858 |