Dataset: heart_disease (classification)
Penalty: 0.1
Seed: 42
Best fitness: -0.6286140185988079
Final val loss: 0.68592216
Final penalty: 0.00836568
Model saved to: exp_1/logs/classification/heart_disease/CONN_LAYER_ADD_PROB/CONN_LAYER_ADD_PROB=0.5/models/best_model_penalty_0.1_seed_42.pth

Final architecture & hyperparameters:
  num_layers: 3
  layer_sizes: [7, 13, 8]
  activations: [3, 3, 3]
  dropout_rates: [0.052, 0.493, 0.101]
  batch_norms: [1, 0, 0]
  learning_rate: 0.0161
  batch_size: 16
  patience: 14
  optimizer_type: 2
  init_type: 0
  l2_penalty: 0.0

Validation metrics (final):
  accuracy: 62.22222222222222
  precision: 31.146853146853143
  recall: 36.16666666666667
  f1_score: 32.05384615384615
  confusion_matrix: [[23, 1, 0, 0, 0], [2, 2, 1, 3, 0], [1, 2, 0, 3, 0], [0, 0, 2, 3, 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: 63.04347826086957
  precision: 23.142857142857146
  recall: 39.2
  f1_score: 27.980861244019138
  confusion_matrix: [[24, 0, 1, 0, 0], [6, 0, 0, 3, 0], [0, 0, 0, 5, 0], [0, 0, 0, 5, 0], [0, 0, 1, 1, 0]]
  num_classes: 5
  class_distribution: {0: 25, 1: 9, 2: 5, 3: 5, 4: 2}
