visualizer.py

import matplotlib.pyplot as plt
import numpy as np


def visualize(layers, weights, activations=[], title='Визуализация Нейронной Сети'):
    """Show given neural network as a plot"""
    for i, layer in enumerate(layers):
        for j in range(layer):
            neuron_x = i
            neuron_y = max(layers) - (max(layers) - layer) / 2 - j  # Распределяем нейроны по вертикали
            if i == 0:
                neuron_color = 'blue'  # Синий во входном слое
            elif i == len(layers) - 1:
                neuron_color = '#FFAA66'  # Бежевый в выходном слое
            else:
                neuron_color = 'gray'  # Серый в скрытых слоях

            if i < len(layers) - 1:
                next_layer = layers[i + 1]
                next_weights = weights[i]
                for k in range(next_layer):
                    next_neuron_x = i + 1
                    next_neuron_y = max(layers) - (max(layers) - next_layer) / 2 - k
                    next_weight = next_weights[j][k]
                    if next_weight > 0:
                        plt.plot([neuron_x, next_neuron_x], [neuron_y, next_neuron_y], color='green', lw=np.arctan(next_weight)*8)
                    elif next_weight < 0:
                        plt.plot([neuron_x, next_neuron_x], [neuron_y, next_neuron_y], color='red', lw=-np.arctan(next_weight)*8)
            
            plt.scatter(neuron_x, neuron_y, color=neuron_color, s=400, zorder=10)  # Рисует один нейрон
            if activations:
                plt.text(neuron_x, neuron_y, activations[i], fontsize=12, ha='center', zorder=11)

    plt.title(title, fontsize=16)
    plt.xlabel('Слои', fontsize=14)
    plt.ylabel('Нейроны', fontsize=14)
    plt.xticks(range(len(layers)), [f'Слой {i}' for i in range(len(layers))], fontsize=12)
    plt.yticks([])
    plt.show()



def main():
    neural_network_layers = [2, 3, 1]  # Входной слой, скрытый слой, выходной слой
    neural_network_weights = [[[6, 2, -1],
                               [-8, 4, 1]],
                               [[3], [7], [-0.5]]]  # Веса модели
    neural_network_activations = ['', '_/', '']  # Функции активации
    visualize(neural_network_layers, neural_network_weights, neural_network_activations)

if __name__ == "__main__":
    main()