Dataset: heart_disease (classification)
Penalty: 0.1
Seed: 42
Best fitness: -0.6261883179220668
Final val loss: 0.65996371
Final penalty: 0.00747283
Model saved to: exp_2/logs/classification/heart_disease/models/best_model_penalty_0.1_seed_42.pth

Final architecture & hyperparameters:
  num_layers: 3
  layer_sizes: [5, 22, 5]
  activations: [3, 2, 3]
  dropout_rates: [0.043, 0.223, 0.043]
  batch_norms: [1, 1, 1]
  learning_rate: 0.0193
  batch_size: 16
  patience: 10
  optimizer_type: 2
  init_type: 0
  l2_penalty: 0.0001

Validation metrics (final):
  accuracy: 62.22222222222222
  precision: 31.80952380952381
  recall: 32.99999999999999
  f1_score: 31.888111888111894
  confusion_matrix: [[24, 0, 0, 0, 0], [3, 2, 2, 1, 0], [1, 2, 0, 3, 0], [0, 0, 3, 2, 0], [0, 1, 1, 0, 0]]
  num_classes: 5
  class_distribution: {0: 24, 1: 8, 2: 6, 3: 5, 4: 2}

Test metrics (final):
  accuracy: 58.69565217391305
  precision: 21.454545454545453
  recall: 31.2
  f1_score: 24.954545454545453
  confusion_matrix: [[24, 1, 0, 0, 0], [6, 0, 0, 3, 0], [0, 1, 0, 4, 0], [0, 1, 1, 3, 0], [0, 1, 0, 1, 0]]
  num_classes: 5
  class_distribution: {0: 25, 1: 9, 2: 5, 3: 5, 4: 2}
