keras-model-viewer
Version:
SVG-based viewer for Keras models
69 lines (46 loc) • 1.54 kB
Markdown
View a Keras Model Object as a nice diagram with enabled zooming.
See example: https://neurals-ro.github.io/keras-model-viewer/examples/index.html
There is also a Keras model editor tool: https://github.com/neurals-ro/keras-model-editor, with an example at https://neurals-ro.github.io/keras-model-editor/examples/index.html
See `examples` folder.
```
npm i keras-model-viewer
```
```
new KerasModelViewer( kerasModelObject, elementId);
new KerasModelViewer( kerasModelObject, document.getElementById(elementId));
new KerasModelViewer( kerasModelObject, document.getElementById(elementId), {
"rankdir": "LR", // "LR" "UD"
"nodesep": 20,
"edgesep": 20,
"ranksep": 40,
"marginx": 0,
"marginy": 0
});
```
```
<div id="kerasmodel"></div>
```
```
let model = new KerasModelViewer( kerasModelObject, 'kerasmodel', {
"rankdir": "LR", // "LR" "UD"
"nodesep": 20,
"edgesep": 20,
"ranksep": 40,
"marginx": 0,
"marginy": 0
});
model.show();
```
`KerasModelViewer.show()` - renders the SVG diagram
`KerasModelViewer.clear()` - removes the SVG diagram
`KerasModelViewer.model([newJSON])` - re-renders the SVG diagram; if new model provided, it renders the new model
- Does not work with all versions of Keras models (does not work with Keras ver. ^2.0.0)
## We would love to know how to improve this
This is in pre-alpha stage but still usable. Let us know what features you would like to see in version 1.
## License
MIT