Dataset: heart_disease (classification)
Penalty: 0.1
Seed: 42
Best fitness: -0.6163784706505185
Final val loss: 0.62350522
Final penalty: 0.01215062
Model saved to: exp_1/logs/classification/heart_disease/LAYER_SIZE_MUTATION_PROB/LAYER_SIZE_MUTATION_PROB=0.4/models/best_model_penalty_0.1_seed_42.pth

Final architecture & hyperparameters:
  num_layers: 4
  layer_sizes: [10, 14, 7, 9]
  activations: [2, 2, 3, 3]
  dropout_rates: [0.28, 0.465, 0.362, 0.031]
  batch_norms: [0, 0, 0, 1]
  learning_rate: 0.0219
  batch_size: 16
  patience: 22
  optimizer_type: 2
  init_type: 0
  l2_penalty: 0.0001

Validation metrics (final):
  accuracy: 60.0
  precision: 24.854700854700855
  recall: 29.166666666666664
  f1_score: 26.11097880447726
  confusion_matrix: [[24, 0, 0, 0, 0], [3, 1, 4, 0, 0], [0, 4, 2, 0, 0], [0, 0, 5, 0, 0], [0, 0, 2, 0, 0]]
  num_classes: 5
  class_distribution: {0: 24, 1: 8, 2: 6, 3: 5, 4: 2}

Test metrics (final):
  accuracy: 63.04347826086957
  precision: 22.666666666666664
  recall: 39.2
  f1_score: 27.454545454545453
  confusion_matrix: [[24, 0, 1, 0, 0], [6, 0, 3, 0, 0], [0, 0, 5, 0, 0], [0, 1, 4, 0, 0], [0, 0, 2, 0, 0]]
  num_classes: 5
  class_distribution: {0: 25, 1: 9, 2: 5, 3: 5, 4: 2}
