File size: 1,517 Bytes
903692d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
{
"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
}
}
} |