47 lines
1.1 KiB
JSON
47 lines
1.1 KiB
JSON
{
|
|
"config": {
|
|
"data_dir": "outlook/data/budget_density_dataset_debug",
|
|
"epochs": 2,
|
|
"batch_size": 2,
|
|
"lr": 0.001,
|
|
"density_weight": 1.0,
|
|
"mono_weight": 0.0,
|
|
"seed": 42
|
|
},
|
|
"per_budget": {
|
|
"budget_1000": {
|
|
"mse": 2.209566116333008,
|
|
"corr": -0.5653519502499024,
|
|
"top20_overlap": 0.037037037037037035,
|
|
"n_graphs": 1
|
|
},
|
|
"budget_2000": {
|
|
"mse": 4.281199932098389,
|
|
"corr": -0.4546216071701198,
|
|
"top20_overlap": 0.07407407407407407,
|
|
"n_graphs": 1
|
|
},
|
|
"overall": {
|
|
"mse": 3.2453832626342773,
|
|
"corr": -0.4747666722392795
|
|
}
|
|
},
|
|
"training_log": [
|
|
{
|
|
"epoch": 0,
|
|
"train_loss": 10.175954818725586,
|
|
"val_loss": 3.5997772216796875,
|
|
"val_corr": -0.5083683560193551,
|
|
"val_top20_overlap": 0.07407407407407407,
|
|
"lr": 0.001
|
|
},
|
|
{
|
|
"epoch": 1,
|
|
"train_loss": 7.313832759857178,
|
|
"val_loss": 3.2453830242156982,
|
|
"val_corr": -0.4747666660841676,
|
|
"val_top20_overlap": 0.05555555555555555,
|
|
"lr": 0.001
|
|
}
|
|
]
|
|
} |