checkpoints/final_metrics.json

{
  "config": {
    "SEED": 42,
    "DETERMINISTIC": false,
    "SPLIT_MODE": "standard",
    "TRAIN_RATIO": 0.8,
    "VAL_RATIO": 0.1,
    "TEST_RATIO": 0.1,
    "EMB_DIM": 128,
    "N_HEADS": 4,
    "N_LAYERS": 4,
    "MLP_HIDDEN": 256,
    "DROPOUT": 0.1,
    "NUM_CLASSES": 10,
    "NUM_STAGES": 3,
    "NUM_RANKS": 13,
    "NUM_SUITS": 4,
    "IMBALANCE_MODE": "sampler",
    "FOCAL_GAMMA": 2.0,
    "FOCAL_ALPHA": null,
    "CLASS_WEIGHT_POWER": 0.5,
    "CLASS_WEIGHT_CAP": 10.0,
    "BATCH_SIZE": 2048,
    "NUM_WORKERS": 2,
    "EPOCHS": 30,
    "PATIENCE": 5,
    "MIN_DELTA": 0.0001,
    "LR": 0.001,
    "WEIGHT_DECAY": 0.001,
    "LR_MIN": 1e-05,
    "GRAD_CLIP": 1.0,
    "USE_STRENGTH_HEAD": true,
    "STRENGTH_LOSS_WEIGHT": 0.5,
    "CACHE_DIR": "cache",
    "USE_CACHE": true,
    "CKPT_DIR": "checkpoints"
  },
  "val": {
    "accuracy": 0.9999866666666667,
    "macro_f1": 0.9847381416701527,
    "per_street_f1": {
      "flop": 1.0,
      "turn": 1.0,
      "river": 0.979994770668576
    },
    "rare_class_recall": {
      "FOUR OF A KIND": 1.0,
      "STRAIGHT FLUSH": 0.9459459459459459,
      "ROYAL FLUSH": 0.8
    }
  },
  "test": {
    "accuracy": 0.9999866666666667,
    "macro_f1": 0.9668325236041202,
    "per_street_f1": {
      "flop": 0.8666666666666666,
      "turn": 0.9589743589743589,
      "river": 0.9818505540864481
    },
    "rare_class_recall": {
      "FOUR OF A KIND": 1.0,
      "STRAIGHT FLUSH": 0.9696969696969697,
      "ROYAL FLUSH": 0.5714285714285714
    }
  }
}