| { |
| "provenance": { |
| "script": "/media/drn2/External/TARA-Oceans/03_analyses/WorldModelApp/scripts/t13_hyperparam_sweep_20260127_130000.py", |
| "inputs": [ |
| "/media/drn2/External/TARA-Oceans/03_analyses/WorldModelApp/data/consolidated_env.npy", |
| "/media/drn2/External/TARA-Oceans/03_analyses/WorldModelApp/data/consolidated_pfam_modules.npy", |
| "/media/drn2/External/TARA-Oceans/03_analyses/WorldModelApp/data/consolidated_bio_response.npy", |
| "/media/drn2/External/TARA-Oceans/03_analyses/WorldModelApp/data/consolidated_bio_valid.npy" |
| ], |
| "date": "2026-01-27 11:11:45", |
| "integrity_check": "PASSED" |
| }, |
| "sweep_config": { |
| "fold": "Atlantic", |
| "n_train": 1166, |
| "n_val": 644, |
| "seeds": [ |
| 42, |
| 123, |
| 456 |
| ], |
| "total_runs": 54, |
| "total_time_s": 214.4 |
| }, |
| "best_config": { |
| "latent_dim": 32, |
| "dropout": 0.3, |
| "vicreg_name": "default", |
| "lambda_inv": 25.0, |
| "lambda_var": 25.0, |
| "lambda_cov": 1.0, |
| "pred_alpha": 1.0, |
| "lr": 0.001, |
| "weight_decay": 0.0001, |
| "max_epochs": 300, |
| "patience": 30, |
| "min_delta": 0.0001, |
| "batch_size": 128, |
| "grad_clip": 1.0 |
| }, |
| "best_metrics": { |
| "avg_val_total": 41.05130301581489, |
| "std_val_total": 0.38005576687680837, |
| "avg_val_vicreg": 38.984246571858726, |
| "std_val_vicreg": 0.3580761158729817, |
| "avg_val_pred": 2.067056621114413, |
| "std_val_pred": 0.09396329900913643, |
| "params": 53187 |
| }, |
| "all_results": [ |
| { |
| "rank": 1, |
| "latent_dim": 32, |
| "dropout": 0.3, |
| "vicreg_name": "default", |
| "lambda_inv": 25.0, |
| "lambda_var": 25.0, |
| "lambda_cov": 1.0, |
| "params": 53187, |
| "avg_val_total": 41.05130301581489, |
| "std_val_total": 0.38005576687680837, |
| "avg_val_vicreg": 38.984246571858726, |
| "std_val_vicreg": 0.3580761158729817, |
| "avg_val_pred": 2.067056621114413, |
| "std_val_pred": 0.09396329900913643 |
| }, |
| { |
| "rank": 2, |
| "latent_dim": 16, |
| "dropout": 0.3, |
| "vicreg_name": "default", |
| "lambda_inv": 25.0, |
| "lambda_var": 25.0, |
| "lambda_cov": 1.0, |
| "params": 47971, |
| "avg_val_total": 41.41749254862467, |
| "std_val_total": 0.1648006392111517, |
| "avg_val_vicreg": 38.50013097127279, |
| "std_val_vicreg": 0.35088072688752703, |
| "avg_val_pred": 2.917361912628015, |
| "std_val_pred": 0.21399294374027156 |
| }, |
| { |
| "rank": 3, |
| "latent_dim": 64, |
| "dropout": 0.3, |
| "vicreg_name": "default", |
| "lambda_inv": 25.0, |
| "lambda_var": 25.0, |
| "lambda_cov": 1.0, |
| "params": 63619, |
| "avg_val_total": 41.53866449991862, |
| "std_val_total": 0.2151686493501874, |
| "avg_val_vicreg": 39.844089084201386, |
| "std_val_vicreg": 0.1480184844259856, |
| "avg_val_pred": 1.6945752816067803, |
| "std_val_pred": 0.10930353765485408 |
| }, |
| { |
| "rank": 4, |
| "latent_dim": 32, |
| "dropout": 0.5, |
| "vicreg_name": "default", |
| "lambda_inv": 25.0, |
| "lambda_var": 25.0, |
| "lambda_cov": 1.0, |
| "params": 53187, |
| "avg_val_total": 42.267934799194336, |
| "std_val_total": 0.40859077604106053, |
| "avg_val_vicreg": 39.89259719848633, |
| "std_val_vicreg": 0.3256019664506181, |
| "avg_val_pred": 2.375337744752566, |
| "std_val_pred": 0.2298646191069577 |
| }, |
| { |
| "rank": 5, |
| "latent_dim": 64, |
| "dropout": 0.5, |
| "vicreg_name": "default", |
| "lambda_inv": 25.0, |
| "lambda_var": 25.0, |
| "lambda_cov": 1.0, |
| "params": 63619, |
| "avg_val_total": 42.66293801201714, |
| "std_val_total": 0.2136225260624326, |
| "avg_val_vicreg": 40.723184373643655, |
| "std_val_vicreg": 0.20533054186665559, |
| "avg_val_pred": 1.939753719088104, |
| "std_val_pred": 0.011867779844547188 |
| }, |
| { |
| "rank": 6, |
| "latent_dim": 16, |
| "dropout": 0.5, |
| "vicreg_name": "default", |
| "lambda_inv": 25.0, |
| "lambda_var": 25.0, |
| "lambda_cov": 1.0, |
| "params": 47971, |
| "avg_val_total": 42.71917851765951, |
| "std_val_total": 0.7665875124356349, |
| "avg_val_vicreg": 39.40584627787272, |
| "std_val_vicreg": 0.6577386783806081, |
| "avg_val_pred": 3.3133317819899983, |
| "std_val_pred": 0.12508679730393615 |
| }, |
| { |
| "rank": 7, |
| "latent_dim": 32, |
| "dropout": 0.3, |
| "vicreg_name": "high_covariance", |
| "lambda_inv": 25.0, |
| "lambda_var": 25.0, |
| "lambda_cov": 10.0, |
| "params": 53187, |
| "avg_val_total": 44.4754458533393, |
| "std_val_total": 0.06584588928393995, |
| "avg_val_vicreg": 42.07578637864855, |
| "std_val_vicreg": 0.28418722198316027, |
| "avg_val_pred": 2.399659642328819, |
| "std_val_pred": 0.24705891220031367 |
| }, |
| { |
| "rank": 8, |
| "latent_dim": 32, |
| "dropout": 0.5, |
| "vicreg_name": "high_covariance", |
| "lambda_inv": 25.0, |
| "lambda_var": 25.0, |
| "lambda_cov": 10.0, |
| "params": 53187, |
| "avg_val_total": 44.576649559868706, |
| "std_val_total": 0.17545108648607916, |
| "avg_val_vicreg": 42.15482881334093, |
| "std_val_vicreg": 0.12596852085536833, |
| "avg_val_pred": 2.421820650084151, |
| "std_val_pred": 0.050243832835822594 |
| }, |
| { |
| "rank": 9, |
| "latent_dim": 16, |
| "dropout": 0.3, |
| "vicreg_name": "high_covariance", |
| "lambda_inv": 25.0, |
| "lambda_var": 25.0, |
| "lambda_cov": 10.0, |
| "params": 47971, |
| "avg_val_total": 44.600736194186744, |
| "std_val_total": 0.16113749177324208, |
| "avg_val_vicreg": 41.29713821411133, |
| "std_val_vicreg": 0.27327076432329706, |
| "avg_val_pred": 3.3035979196429253, |
| "std_val_pred": 0.25611801797899636 |
| }, |
| { |
| "rank": 10, |
| "latent_dim": 16, |
| "dropout": 0.5, |
| "vicreg_name": "high_covariance", |
| "lambda_inv": 25.0, |
| "lambda_var": 25.0, |
| "lambda_cov": 10.0, |
| "params": 47971, |
| "avg_val_total": 44.66786532931857, |
| "std_val_total": 0.2543020078561328, |
| "avg_val_vicreg": 41.54936239454481, |
| "std_val_vicreg": 0.13479661347127272, |
| "avg_val_pred": 3.1185029836164575, |
| "std_val_pred": 0.1669271848740466 |
| }, |
| { |
| "rank": 11, |
| "latent_dim": 64, |
| "dropout": 0.3, |
| "vicreg_name": "high_covariance", |
| "lambda_inv": 25.0, |
| "lambda_var": 25.0, |
| "lambda_cov": 10.0, |
| "params": 63619, |
| "avg_val_total": 44.759677463107636, |
| "std_val_total": 0.13176357883806153, |
| "avg_val_vicreg": 42.70491578843859, |
| "std_val_vicreg": 0.04728248324726026, |
| "avg_val_pred": 2.054761736343304, |
| "std_val_pred": 0.1130719732622441 |
| }, |
| { |
| "rank": 12, |
| "latent_dim": 64, |
| "dropout": 0.5, |
| "vicreg_name": "high_covariance", |
| "lambda_inv": 25.0, |
| "lambda_var": 25.0, |
| "lambda_cov": 10.0, |
| "params": 63619, |
| "avg_val_total": 45.132680045233826, |
| "std_val_total": 0.05655242970803006, |
| "avg_val_vicreg": 43.10596889919705, |
| "std_val_vicreg": 0.057764934163928965, |
| "avg_val_pred": 2.026710879057646, |
| "std_val_pred": 0.06620457474314766 |
| }, |
| { |
| "rank": 13, |
| "latent_dim": 32, |
| "dropout": 0.3, |
| "vicreg_name": "high_variance", |
| "lambda_inv": 10.0, |
| "lambda_var": 50.0, |
| "lambda_cov": 1.0, |
| "params": 53187, |
| "avg_val_total": 54.442194832695854, |
| "std_val_total": 0.3233563313102728, |
| "avg_val_vicreg": 52.32098049587674, |
| "std_val_vicreg": 0.3367353449524611, |
| "avg_val_pred": 2.1212142213351197, |
| "std_val_pred": 0.1534630946054846 |
| }, |
| { |
| "rank": 14, |
| "latent_dim": 16, |
| "dropout": 0.3, |
| "vicreg_name": "high_variance", |
| "lambda_inv": 10.0, |
| "lambda_var": 50.0, |
| "lambda_cov": 1.0, |
| "params": 47971, |
| "avg_val_total": 54.65630997551813, |
| "std_val_total": 1.5158442768219174, |
| "avg_val_vicreg": 51.99490547180176, |
| "std_val_vicreg": 1.424356508362817, |
| "avg_val_pred": 2.6614045856727495, |
| "std_val_pred": 0.12813470575119099 |
| }, |
| { |
| "rank": 15, |
| "latent_dim": 64, |
| "dropout": 0.3, |
| "vicreg_name": "high_variance", |
| "lambda_inv": 10.0, |
| "lambda_var": 50.0, |
| "lambda_cov": 1.0, |
| "params": 63619, |
| "avg_val_total": 58.833312564425995, |
| "std_val_total": 0.684309109473459, |
| "avg_val_vicreg": 56.4467887878418, |
| "std_val_vicreg": 0.7851219132682428, |
| "avg_val_pred": 2.3865235398213067, |
| "std_val_pred": 0.10826007102064247 |
| }, |
| { |
| "rank": 16, |
| "latent_dim": 32, |
| "dropout": 0.5, |
| "vicreg_name": "high_variance", |
| "lambda_inv": 10.0, |
| "lambda_var": 50.0, |
| "lambda_cov": 1.0, |
| "params": 53187, |
| "avg_val_total": 64.88169436984593, |
| "std_val_total": 0.4121604408754495, |
| "avg_val_vicreg": 62.20666758219401, |
| "std_val_vicreg": 0.25507370131286555, |
| "avg_val_pred": 2.6750266907943625, |
| "std_val_pred": 0.19599384353880261 |
| }, |
| { |
| "rank": 17, |
| "latent_dim": 16, |
| "dropout": 0.5, |
| "vicreg_name": "high_variance", |
| "lambda_inv": 10.0, |
| "lambda_var": 50.0, |
| "lambda_cov": 1.0, |
| "params": 47971, |
| "avg_val_total": 66.12008539835612, |
| "std_val_total": 1.593290289328216, |
| "avg_val_vicreg": 62.57050938076443, |
| "std_val_vicreg": 1.6494563418144146, |
| "avg_val_pred": 3.549575644234816, |
| "std_val_pred": 0.07241669466839466 |
| }, |
| { |
| "rank": 18, |
| "latent_dim": 64, |
| "dropout": 0.5, |
| "vicreg_name": "high_variance", |
| "lambda_inv": 10.0, |
| "lambda_var": 50.0, |
| "lambda_cov": 1.0, |
| "params": 63619, |
| "avg_val_total": 68.17775344848633, |
| "std_val_total": 0.332835219927688, |
| "avg_val_vicreg": 65.75349447462293, |
| "std_val_vicreg": 0.4273214717910489, |
| "avg_val_pred": 2.4242580677900047, |
| "std_val_pred": 0.10955952723640604 |
| } |
| ] |
| } |
