NeuralNetworkVisualizer/config.py at main · cryybash/NeuralNetworkVisualizer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
# Config file for Neural Network visualizer
# Modify these values to customize the appearance and behavior

import pygame

# Window Settings
WINDOW_WIDTH = 1600
WINDOW_HEIGHT = 800
FPS = 60
BACKGROUND_COLOR = (15, 15, 25)  # Dark blue-black

# Network Visualization Settings
NETWORK_PANEL_WIDTH = 1000
NETWORK_PANEL_X = 0
NETWORK_PANEL_Y = 0

# Stats Panel Settings
STATS_PANEL_WIDTH = WINDOW_WIDTH - NETWORK_PANEL_WIDTH
STATS_PANEL_X = NETWORK_PANEL_WIDTH
STATS_PANEL_Y = 0
STATS_PANEL_COLOR = (25, 25, 35)

# Neuron Settings
NEURON_RADIUS = 16
NEURON_BORDER_WIDTH = 2
NEURON_SPACING_X = 200  # Horizontal spacing between layers
NEURON_SPACING_Y = 30   # Vertical spacing between neurons
NEURON_INACTIVE_COLOR = (40, 40, 60)
NEURON_ACTIVE_COLOR = (255, 100, 100)  # Red when fully activated
NEURON_BORDER_COLOR = (100, 100, 120)

# Connection Settings
CONNECTION_INACTIVE_COLOR = (50, 50, 70)
CONNECTION_ACTIVE_COLOR = (100, 200, 255)  # Cyan when active
CONNECTION_MIN_ALPHA = 30
CONNECTION_MAX_ALPHA = 180   # 300
CONNECTION_WIDTH = 2

# Animation Settings
ACTIVATION_SPEED = 0.5  # How fast neurons pulse (0-1) 0.1
GLOW_ENABLED = True
GLOW_RADIUS = 5
PARTICLE_EFFECTS = True  # want to add in the future

# Text Settings
FONT_SIZE_LARGE = 28
FONT_SIZE_MEDIUM = 20
FONT_SIZE_SMALL = 16
TEXT_COLOR = (200, 200, 220)
TITLE_COLOR = (255, 255, 255)

# Training Settings
TRAIN_UPDATE_FREQUENCY = 5  # Update visualization every N epochs
GRAPH_HISTORY_LENGTH = 100  # How many epochs to show in graphs

# Color Schemes
COLOR_SCHEME = "CYBER"

COLOR_SCHEMES = {
    "CYBER": {
        "bg": (15, 15, 25),
        "neuron_inactive": (40, 40, 60),
        "neuron_active": (255, 100, 100),
        "connection": (100, 200, 255),
        "accent": (0, 255, 200)
    },
    "MATRIX": {
        "bg": (0, 10, 0),
        "neuron_inactive": (0, 40, 0),
        "neuron_active": (0, 255, 0),
        "connection": (0, 150, 0),
        "accent": (0, 255, 100)
    },
    "FIRE": {
        "bg": (20, 10, 5),
        "neuron_inactive": (60, 30, 20),
        "neuron_active": (255, 100, 0),
        "connection": (255, 150, 50),
        "accent": (255, 200, 0)
    },
    "ICE": {
        "bg": (10, 15, 25),
        "neuron_inactive": (30, 50, 80),
        "neuron_active": (100, 200, 255),
        "connection": (150, 220, 255),
        "accent": (200, 240, 255)
    }
}

# Get current color scheme
def get_color(key):
    return COLOR_SCHEMES[COLOR_SCHEME].get(key, (255, 255, 255))