Unggi's picture
leap-kt: 1 dataset(s), 23 files
d27e706 verified
Raw
History Blame Contribute Delete
1.52 kB
{
"model_type": "knowledgequerynetworkmod",
"dataset": "assist2009",
"n_folds_completed": 5,
"n_folds_failed": 0,
"fold_failures": [],
"leap_kt_version": "0.1.0",
"hardware": {
"device": "cuda",
"accelerator": "NVIDIA GeForce RTX 3090",
"torch_version": "2.9.1+cu128",
"cuda_version": "12.8",
"cpu": "x86_64",
"platform": "Linux 5.15.0-140-generic"
},
"protocol": {
"test_ratio": 0.2,
"n_folds": 5,
"early_stopping_patience": 10,
"early_stopping_metric": "auc",
"max_epochs": 200,
"primary_metric": "auc",
"secondary_metrics": [
"acc",
"f1"
],
"validation_source": "held_out_fold",
"seed": 0,
"is_deviation": false,
"deviation_reason": ""
},
"auc_mean": 0.7330049650742498,
"auc_std": 0.0028354218157567303,
"auc_per_fold": [
0.7309179251115937,
0.7372997864201055,
0.730040247253478,
0.7337584087272122,
0.7330084578588602
],
"acc_mean": 0.7196094914218027,
"acc_std": 0.006821990035265488,
"acc_per_fold": [
0.7093729700813878,
0.7233197050170036,
0.7168048603415995,
0.7271216231706851,
0.7214282984983379
],
"f1_mean": 0.7995526525669027,
"f1_std": 0.009653044308856292,
"f1_per_fold": [
0.7846179985274961,
0.8045772272150703,
0.7955025177622956,
0.8088335675567155,
0.8042319517729354
],
"batch_size": 32,
"wall_seconds": 48700.2,
"git_commit": "0876af0d9bc70258b56b6685eaf4e41c98d6ee10",
"reference_auc": null
}