一区top轴承故障诊断迁移学习代码复现
故障诊断代码 复现
首先使用一维的cnn对源域和目标域进行特征提取，域适应阶段：将源域和目标域作为cnn的输入得到特征，然后进行边缘概率分布对齐和条件概率分布对齐，也就是进行JDA联合对齐。此域适应方法特别适合初学者，了解迁移学习的基础知识。
数据预处理：1维数据
网络模型：1D-CNN-MMD-Coral
数据集：西储大学CWRU
准确率：99%
网络框架：pytorch
结果输出：损失曲线图、准确率曲线图、混淆矩阵、tsne图
使用对象：初学者
注意：此代码是一个在 GPU 上跑的代码，有宝子的电脑只支持 cpu，只需要将代码修改成只在 cpu 上跑的就行
YID:4115962570899045
北方做DIY的河豚
