% 导入数据集
load(‘fault_diagnosis_data.mat’); % 假设故障诊断数据保存在fault_diagnosis_data.mat文件中
% 数据预处理
% 这里省略了数据预处理的步骤，包括数据归一化、特征提取等
% 划分训练集和测试集
train_ratio = 0.8; % 训练集占总数据的比例
train_size = round(train_ratio * size(data, 1));
train_data = data(1:train_size,