File size: 1,767 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: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
```json
{
  "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
```json
{
  "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