Plugin for visual display of a neural network.
Project description
jupyter: kernelspec: display_name: Python 3 language: python name: python3 language_info: codemirror_mode: name: ipython version: 2 file_extension: .py mimetype: text/x-python name: python nbconvert_exporter: python pygments_lexer: ipython2 version: 2.7.6 nbformat: 4 nbformat_minor: 5
pip install neural-net-drawer
import matplotlib.pyplot as plt
from neural-net-drawer.drawer import NeuralNetworkDiagram()
# Пример использования
fig = plt.figure(figsize=(9, 9))
ax = fig.gca()
ax.axis('off')
# Создание экземпляра класса
nn_diagram = NeuralNetworkDiagram()
# Вызов метода для построения диаграммы нейронной сети
nn_diagram.draw_neural_net(ax, [7, 5, 4, 3, 4, 2, 1])
plt.show()
# Пример использования
fig = plt.figure(figsize=(9, 9))
ax = fig.gca()
ax.axis('off')
# Создание экземпляра класса
nn_diagram = NeuralNetworkDiagram()
#Максимальное количество отображаемых нейронов в сети
nn_diagram.max_n_layers_size = 3
#Максимальное количество отображаемых нейронов в сети.
nn_diagram.max_layer_size = 10
# Вызов метода для построения диаграммы нейронной сети
nn_diagram.draw_neural_net(ax, [7, 5, 4, 3, 4, 2, 1])
plt.show()
fig = plt.figure(figsize=(9, 9))
ax = fig.gca()
ax.axis('off')
# Создание экземпляра класса
nn_diagram = NeuralNetworkDiagram()
#Максимальное количество отображаемых нейронов в сети
nn_diagram.max_n_layers_size = 3
nn_diagram.show_neuron_numbers = False
# Вызов метода для построения диаграммы нейронной сети
nn_diagram.draw_neural_net(ax, [7, 5, 4, 3, 4, 2, 1])
plt.show()
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.