轴承故障诊断python代码

from keras.layers import Conv1D Dense Dropout BatchNormalization MaxPooling1D Activation Flatten
from keras.models import Sequential
from keras.utils import plot_model
from keras.regularizers import l2
import preprocess
from keras.callbacks import TensorBoard
import numpy as np

# 训练参数
batch_size = 128
epochs = 20
num_classes = 10
length = 2048
BatchNorm = True # 是否批量归一化
number = 1000 # 每类样本的数量
normal = True # 是否标准化
rate = [0.70.20.1] # 测试集验证集划分比例

path = r‘data\0HP‘
x_train y_train x_valid y_valid x_test y_test = preprocess.prepro（d_path=pathlength=length
                                                                  number=number
                                                                  normal=normal
                                                                  rate=rate
                                                                  enc=True enc_step=28）
# 输入卷积的时候还需要修改一下，增加通道数目
x_train x_valid x_test = x_train[::np.newaxis] x_valid[::np.new