neural-net-drawer 0.0.2

Plugin for visual display of a neural network.

---

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()