{ "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 } ] }