Sph3inxz commited on
Commit
7f9a40c
·
verified ·
1 Parent(s): 9620d36

Add best C-MAPSS JEPA, critic, and RUL checkpoints

Browse files
README.md ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - predictive-maintenance
5
+ - cmapss
6
+ - remaining-useful-life
7
+ - graph-neural-networks
8
+ - jepa
9
+ - anomaly-detection
10
+ library_name: pytorch
11
+ ---
12
+
13
+ # Diagnostic Graph PDM — Best C-MAPSS Checkpoints
14
+
15
+ Best observed checkpoints from the [Diagnostic-Graph-PDM](https://github.com/Sph3inz/Diagnostic-Graph-PDM) stack: Graph-JEPA pretrain → Energy Critic → TCN+Transformer RUL with energy fusion.
16
+
17
+ ## Files
18
+
19
+ | File | Role | Selection criterion |
20
+ |------|------|-------------------|
21
+ | `checkpoints/jepa/jepa_epoch0003.pt` | Graph-JEPA encoder (FD123 + unlabeled FD004) | Epoch 3 val loss 0.0025 |
22
+ | `checkpoints/critic/vulcan_epoch0030.pt` | Energy Critic + encoder (JEPA-init) | Best saved FD1234 critic (~0.97 val F1) |
23
+ | `checkpoints/rul/rul_sequence.pt` | TCN+Transformer RUL (FD001) | Best val RMSE epoch 54 — test RMSE **12.485**, NASA **294.6** |
24
+
25
+ See `checkpoints.json` for paths and metrics.
26
+
27
+ ## Usage
28
+
29
+ ```powershell
30
+ pip install torch typer omegaconf rich
31
+ git clone https://github.com/Sph3inz/Diagnostic-Graph-PDM.git
32
+ cd Diagnostic-Graph-PDM
33
+ pip install -e .
34
+
35
+ hf download Sph3inxz/Diagnostic-Graph-PDM --local-dir ./hf_ckpt
36
+
37
+ # Scan / localize (critic)
38
+ python -m vulcan scan data/cmapss_fd001.graphs.jsonl `
39
+ --checkpoint hf_ckpt/checkpoints/critic/vulcan_epoch0030.pt --format summary
40
+
41
+ # RUL eval
42
+ python eval_ckpt.py hf_ckpt/checkpoints/rul/rul_sequence.pt
43
+ ```
44
+
45
+ Train critic from JEPA:
46
+
47
+ ```powershell
48
+ python -m vulcan train --config configs/critic_cmapss_fd1234_jepa.yaml `
49
+ --resume-from hf_ckpt/checkpoints/jepa/jepa_epoch0003.pt
50
+ ```
51
+
52
+ ## Citation
53
+
54
+ If you use these weights, link the GitHub repo: https://github.com/Sph3inz/Diagnostic-Graph-PDM
checkpoints.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "stack": "jepa-critic-rul",
3
+ "benchmark": "NASA C-MAPSS",
4
+ "checkpoints": {
5
+ "jepa": {
6
+ "path": "checkpoints/jepa/jepa_epoch0003.pt",
7
+ "config": "configs/jepa_cmapss_fd123_plus_fd004_unlabeled.yaml",
8
+ "epoch": 3,
9
+ "notes": "FD123 + unlabeled FD004 JEPA pretrain"
10
+ },
11
+ "critic": {
12
+ "path": "checkpoints/critic/vulcan_epoch0030.pt",
13
+ "config": "configs/critic_cmapss_fd1234_jepa.yaml",
14
+ "epoch": 30,
15
+ "val_f1_approx": 0.9697,
16
+ "notes": "JEPA-initialized FD1234 energy critic; use for scan and RUL energy fusion"
17
+ },
18
+ "rul": {
19
+ "path": "checkpoints/rul/rul_sequence.pt",
20
+ "config": "configs/rul_sota_cmapss.yaml",
21
+ "best_epoch": 54,
22
+ "test_rmse": 12.485,
23
+ "test_mae": 8.963,
24
+ "nasa_score": 294.6,
25
+ "notes": "FD001 TCN+Transformer RUL, best validation checkpoint reloaded"
26
+ }
27
+ },
28
+ "github": "https://github.com/Sph3inz/Diagnostic-Graph-PDM"
29
+ }
checkpoints/critic/vulcan_epoch0030.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e6ebf4c66d20a72f62f1bb945b9ceaff16c290b2d23c450e6ceddb6ce196eb13
3
+ size 9647345
checkpoints/jepa/jepa_epoch0003.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4781ead9e45ab291143ffe9521b74c54d1526ac56a4186fee32caafcc4f844cf
3
+ size 9316120
checkpoints/rul/rul_sequence.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d3cc35c04fd9a8066e24a19cc86baf55985ada7700aab5b751b524686c5badfb
3
+ size 2910535