前言

本文基于凱斯西儲(chǔ)大學(xué)（CWRU）軸承數(shù)據(jù)，先經(jīng)過(guò)數(shù)據(jù)預(yù)處理進(jìn)行數(shù)據(jù)集的制作和加載，最后通過(guò)Pytorch實(shí)現(xiàn)1DCNN-2DResNet并行模型對(duì)故障數(shù)據(jù)的分類(lèi)。

1 軸承故障數(shù)據(jù)的預(yù)處理

1.1 導(dǎo)入數(shù)據(jù)

參考之前的文章，進(jìn)行故障10分類(lèi)的預(yù)處理，凱斯西儲(chǔ)大學(xué)軸承數(shù)據(jù)10分類(lèi)數(shù)據(jù)集：

train_set、val_set、test_set 均為按照7：2：1劃分訓(xùn)練集、驗(yàn)證集、測(cè)試集，最后保存數(shù)據(jù)

上圖是數(shù)據(jù)的讀取形式以及預(yù)處理思路

1.2 數(shù)據(jù)預(yù)處理，制作數(shù)據(jù)集

2 基于Pytorch的1DCNN-2DResNet的軸承故障診斷

2.1 定義1DCNN-2DResNet分類(lèi)網(wǎng)絡(luò)模型

2.2 設(shè)置參數(shù)，訓(xùn)練模型

50個(gè)epoch，準(zhǔn)確率100%，用1DCNN-2DResNet并行網(wǎng)絡(luò)分類(lèi)效果顯著，模型能夠充分提取軸承故障信號(hào)的全局空間、時(shí)頻域和局部特征，收斂速度快，性能優(yōu)越，精度高，效果明顯!

2.3 模型評(píng)估

準(zhǔn)確率、精確率、召回率、F1 Score

故障十分類(lèi)混淆矩陣：

點(diǎn)擊下載：原文完整數(shù)據(jù)、Python代碼

???https://mbd.pub/o/bread/ZpWakplp????

本文轉(zhuǎn)載自 建模先鋒，作者： 小蝸愛(ài)建模