Dataset: heart_disease (classification)
Penalty: 0.1
Seed: 42
Best fitness: -0.6073991998226276
Final val loss: 0.64555674
Final penalty: 0.00966615
Model saved to: exp_1/logs/classification/heart_disease/LAYER_SIZE_MUTATION_PROB/LAYER_SIZE_MUTATION_PROB=0.3/models/best_model_penalty_0.1_seed_42.pth

Final architecture & hyperparameters:
  num_layers: 3
  layer_sizes: [6, 15, 12]
  activations: [3, 1, 2]
  dropout_rates: [0.14, 0.32, 0.18]
  batch_norms: [1, 0, 1]
  learning_rate: 0.0229
  batch_size: 64
  patience: 30
  optimizer_type: 2
  init_type: 0
  l2_penalty: 0.0001

Validation metrics (final):
  accuracy: 64.44444444444444
  precision: 31.11111111111111
  recall: 43.33333333333333
  f1_score: 34.80392156862745
  confusion_matrix: [[22, 2, 0, 0, 0], [2, 2, 0, 4, 0], [0, 5, 0, 1, 0], [0, 0, 0, 5, 0], [0, 0, 0, 2, 0]]
  num_classes: 5
  class_distribution: {0: 24, 1: 8, 2: 6, 3: 5, 4: 2}

Test metrics (final):
  accuracy: 56.52173913043478
  precision: 21.428571428571427
  recall: 33.6
  f1_score: 25.02482621648461
  confusion_matrix: [[22, 2, 0, 1, 0], [6, 0, 0, 3, 0], [0, 1, 0, 4, 0], [0, 1, 0, 4, 0], [0, 0, 0, 2, 0]]
  num_classes: 5
  class_distribution: {0: 25, 1: 9, 2: 5, 3: 5, 4: 2}
