Training Provenance
- Date: 2026-01-25 18:31:21
- Script: /scratch/drn2/PROJECTS/TARA-LA4SR/MANUSCRIPT/scripts/train_manifold_20260125_173853.py
- Model: full
- Data: /scratch/drn2/PROJECTS/TARA-LA4SR/MANUSCRIPT/data/manifold_train_val_test_20260125_182033.npz
- Best epoch: 24
- Best val loss: 0.009360
Config
{
"model_type": "full",
"input_dim": 9611,
"output_dim": 64,
"batch_size": 32,
"epochs": 200,
"learning_rate": 0.001,
"weight_decay": 0.0001,
"loss_type": "mse",
"loss_alpha": 0.8,
"scheduler_type": "cosine",
"patience": 30,
"device": "cuda",
"use_amp": true,
"seed": 42
}
Final Metrics
{
"train": {
"loss": 0.009715628852559761,
"mse": 0.009715628852559761,
"cosine_loss": 0.3846368572928689,
"r2": -8.52346420288086e-05,
"mae": 0.07396243512630463,
"cosine_similarity": 0.6160224080085754,
"dim_r2_mean": -8.874572813510895e-05,
"dim_r2_min": -0.0004864931106567383,
"dim_r2_max": -1.1920928955078125e-07
},
"val": {
"loss": 0.00936005897819996,
"mse": 0.00936005897819996,
"cosine_loss": 0.3663846731185913,
"r2": -0.006044507026672363,
"mae": 0.07282450050115585,
"cosine_similarity": 0.6294302940368652,
"dim_r2_mean": -0.006076883524656296,
"dim_r2_min": -0.048920631408691406,
"dim_r2_max": -2.5033950805664062e-06
},
"test": {
"loss": 0.009311655163764953,
"mse": 0.009311655163764953,
"cosine_loss": 0.36379417181015017,
"r2": -0.0057315826416015625,
"mae": 0.07238934934139252,
"cosine_similarity": 0.6390543580055237,
"dim_r2_mean": -0.00670330785214901,
"dim_r2_min": -0.032412052154541016,
"dim_r2_max": -2.9802322387695312e-06
}
}
Integrity Check: PASSED
