Skip to content

Commit 84c506b

Browse files
committed
Add GIP-MI reliability framework example
Signed-off-by: niuchaoyu <411654689@qq.com>
1 parent 764305f commit 84c506b

56 files changed

Lines changed: 44361 additions & 0 deletions

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.
Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
# GCN-Informer
2+
3+
本目录包含 GCN-Informer 的训练/验证/测试代码,以及配套的 CPU / Memory / Response Time 数据集样例。
4+
5+
## 目录结构
6+
7+
- `train_gcn_informer.py`:训练/验证/测试入口(默认使用 `data/myData/2024-05-5_train_constant.csv`
8+
- `data/myData/`:数据集 CSV
9+
- `models/`:模型组件(Informer + GCN)
10+
- `utils/`:mask 与指标计算
11+
- `checkpoints/`:默认保存 `checkpoints/gcn_informer/best_model.pt`
12+
- `results/`:默认输出 `results/gcn_informer/test_predictions.csv`
13+
14+
## 安装依赖
15+
16+
```bash
17+
pip install -r requirements.txt
18+
```
19+
20+
## 运行
21+
22+
默认训练(结束后会跑 test 并导出预测 CSV):
23+
24+
```bash
25+
python train_gcn_informer.py
26+
```
27+
28+
指定数据文件:
29+
30+
```bash
31+
python train_gcn_informer.py --data_path data/myData/2024-07-08_train_non-constant.csv
32+
```
Binary file not shown.
Binary file not shown.

reliability/GIP-MI-demo/GCN-Informer/data/myData/2024-05-5_aging_constant.csv

Lines changed: 10635 additions & 0 deletions
Large diffs are not rendered by default.

reliability/GIP-MI-demo/GCN-Informer/data/myData/2024-05-5_train_constant.csv

Lines changed: 9390 additions & 0 deletions
Large diffs are not rendered by default.

reliability/GIP-MI-demo/GCN-Informer/data/myData/2024-07-08_aging_non-constant.csv

Lines changed: 7120 additions & 0 deletions
Large diffs are not rendered by default.

reliability/GIP-MI-demo/GCN-Informer/data/myData/2024-07-08_train_non-constant.csv

Lines changed: 6218 additions & 0 deletions
Large diffs are not rendered by default.

reliability/GIP-MI-demo/GCN-Informer/models/__init__.py

Whitespace-only changes.
Binary file not shown.
Binary file not shown.

0 commit comments

Comments
 (0)