File size: 1,770 Bytes
df6ede9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
# Training Provenance

- Date: 2026-01-25 18:32:22
- Script: /scratch/drn2/PROJECTS/TARA-LA4SR/MANUSCRIPT/scripts/train_manifold_20260125_173853.py
- Model: light
- Data: /scratch/drn2/PROJECTS/TARA-LA4SR/MANUSCRIPT/data/manifold_train_val_test_20260125_182033.npz
- Best epoch: 56
- Best val loss: 0.009357

## Config
```json
{
  "model_type": "light",
  "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
```json
{
  "train": {
    "loss": 0.009701163846660744,
    "mse": 0.009701163846660744,
    "cosine_loss": 0.38389346545392816,
    "r2": -0.00014269351959228516,
    "mae": 0.07398466765880585,
    "cosine_similarity": 0.6159994006156921,
    "dim_r2_mean": -0.0001436471939086914,
    "dim_r2_min": -0.0009844303131103516,
    "dim_r2_max": -1.1920928955078125e-07
  },
  "val": {
    "loss": 0.009356671385467053,
    "mse": 0.009356671385467053,
    "cosine_loss": 0.3662286400794983,
    "r2": -0.0056684017181396484,
    "mae": 0.07284688949584961,
    "cosine_similarity": 0.6295995712280273,
    "dim_r2_mean": -0.005823422223329544,
    "dim_r2_min": -0.044977784156799316,
    "dim_r2_max": -1.0728836059570312e-06
  },
  "test": {
    "loss": 0.009309758059680462,
    "mse": 0.009309758059680462,
    "cosine_loss": 0.36371852159500123,
    "r2": -0.005517482757568359,
    "mae": 0.07240588963031769,
    "cosine_similarity": 0.6391298174858093,
    "dim_r2_mean": -0.00648270919919014,
    "dim_r2_min": -0.03397870063781738,
    "dim_r2_max": -9.5367431640625e-07
  }
}
```

## Integrity Check: PASSED