├── README.md ├── mat ├── data_C_k3.mat └── data_X_k3.mat ├── models ├── ChebyNet.py ├── GAT.py ├── GCN.py ├── GraphSage.py ├── __init__.py └── __pycache__ │ ├── ChebyNet.cpython-37.pyc │ ├── ChebyNet.cpython-38.pyc │ ├── GAT.cpython-37.pyc │ ├── GAT.cpython-38.pyc │ ├── GATv2.cpython-37.pyc │ ├── GCN.cpython-37.pyc │ ├── GCN.cpython-38.pyc │ ├── GIN.cpython-37.pyc │ ├── GIN.cpython-38.pyc │ ├── GraphNet.cpython-37.pyc │ ├── GraphSage.cpython-37.pyc │ ├── GraphSage.cpython-38.pyc │ ├── HoGCN.cpython-37.pyc │ ├── HoGCN.cpython-38.pyc │ ├── MLP.cpython-37.pyc │ ├── MLP.cpython-38.pyc │ ├── MyNet.cpython-37.pyc │ ├── SGCN.cpython-37.pyc │ ├── SGCN.cpython-38.pyc │ ├── __init__.cpython-37.pyc │ └── __init__.cpython-38.pyc ├── train_graph_diagnosis.py └── utils ├── Set_division.py ├── __init__.py ├── __pycache__ ├── Set_division.cpython-37.pyc ├── __init__.cpython-37.pyc ├── __init__.cpython-38.pyc ├── freeze.cpython-37.pyc ├── freeze.cpython-38.pyc ├── logger.cpython-37.pyc ├── logger.cpython-38.pyc ├── save.cpython-37.pyc ├── save.cpython-38.pyc ├── train_graph_utils.cpython-37.pyc └── train_graph_utils.cpython-38.pyc ├── logger.py ├── save.py └── train_graph_utils.py /README.md: -------------------------------------------------------------------------------- 1 | # KNN-SA-GNN 2 | 论文“时变转速下基于改进图注意力网络的轴承半监督故障诊断”源码 3 | -------------------------------------------------------------------------------- /mat/data_C_k3.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/mat/data_C_k3.mat -------------------------------------------------------------------------------- /mat/data_X_k3.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/mat/data_X_k3.mat -------------------------------------------------------------------------------- /models/ChebyNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/ChebyNet.py -------------------------------------------------------------------------------- /models/GAT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/GAT.py -------------------------------------------------------------------------------- /models/GCN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/GCN.py -------------------------------------------------------------------------------- /models/GraphSage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/GraphSage.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/__pycache__/ChebyNet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/ChebyNet.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/ChebyNet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/ChebyNet.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/GAT.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/GAT.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/GAT.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/GAT.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/GATv2.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/GATv2.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/GCN.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/GCN.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/GCN.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/GCN.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/GIN.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/GIN.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/GIN.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/GIN.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/GraphNet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/GraphNet.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/GraphSage.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/GraphSage.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/GraphSage.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/GraphSage.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/HoGCN.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/HoGCN.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/HoGCN.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/HoGCN.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/MLP.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/MLP.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/MLP.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/MLP.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/MyNet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/MyNet.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/SGCN.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/SGCN.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/SGCN.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/SGCN.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/models/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /train_graph_diagnosis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/train_graph_diagnosis.py -------------------------------------------------------------------------------- /utils/Set_division.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/utils/Set_division.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/python 2 | # -*- coding:utf-8 -*- 3 | -------------------------------------------------------------------------------- /utils/__pycache__/Set_division.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/utils/__pycache__/Set_division.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/utils/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/utils/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/freeze.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/utils/__pycache__/freeze.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/freeze.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/utils/__pycache__/freeze.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/logger.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/utils/__pycache__/logger.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/logger.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/utils/__pycache__/logger.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/save.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/utils/__pycache__/save.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/save.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/utils/__pycache__/save.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/train_graph_utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/utils/__pycache__/train_graph_utils.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/train_graph_utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/utils/__pycache__/train_graph_utils.cpython-38.pyc -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/utils/save.py -------------------------------------------------------------------------------- /utils/train_graph_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yanshen210/KNN-SA-GNN/HEAD/utils/train_graph_utils.py --------------------------------------------------------------------------------