当前位置: 首页 > 创领中心 > 网络优化

一同聊聊图卷积缺点诊断 新增GAT SGCN GIN分类模型

  • 网络优化
  • 2024-11-14

前言

本文基于凯斯西储大学(CWRU)轴承数据,先经过数据预解决启动数据集的制造和加载,最后经过Pytorch成功K-NN+GCN模型对缺点数据的分类。

1.关系网络引见

1.1 图卷积神经网络(GCN)

论文地址:​ ​​ ​

1.2图留意力网络(GAT)

论文地址:​ ​​ ​

1.3稠密图卷积网络(SGCN)

论文地址:​​

1.4GIN 图同构网络

2轴承缺点数据的预解决

2.1 导入数据

参考之前的文章,启动缺点10分类的预解决,凯斯西储大学轴承数据10分类数据集:

train_set、val_set、test_set 均为依照7:2:1划分训练集、验证集、测试集,最后保留数据

上图是数据的读取方式以及预解决思绪

2.2数据预解决,制造数据集

3基于Pytorch的GCN轴承缺点诊断

3.1 定义GCN分类网络模型

3.2 设置参数,训练模型

50个epoch,准确率95%,用GCN网络分类成果清楚,模型能够充沛提取轴承缺点信号的全局空间和部分特色,收敛速度快,性能优越,精度高,成果清楚!

3.3 模型评价

准确率、准确率、召回率、F1 Score

缺点十分类混杂矩阵:

本文转载自​​,作者:

  • 关注微信

本网站的文章部分内容可能来源于网络和网友发布,仅供大家学习与参考,如有侵权,请联系站长进行删除处理,不代表本网站立场,转载联系作者并注明出处:https://clwxseo.com/wangluoyouhua/5994.html

猜你喜欢

热门资讯

关注我们

微信公众号