随机森林模型构建测试–在凯斯西储(CWRU)数据上的简单测试(1)
1. 实验数据集说明
凯斯西储轴承故障数据集是凯斯西储大学制作的一个球轴承故障诊断公开数据集[1]。电机轴承使用电火花加工 (EDM) 植入故障。在内滚道、滚动体(即球)和外滚道处分别引入了直径从0.007英寸到0.040英寸不等的故障。将故障轴承重新安装到测试电机中,并记录电机负载为 0 至 3 马力(电机速度为 1797 至 1720 RPM)的振动加速度传感器监测数据。在测点布置上,多数采集了三个位置的数据,DE表示驱动端,FE表示风扇端,BA表示基座端。
2. 测试数据说明
测试数据为特征提取后数据。前面9列字段为特征,最后一个字段为故障类型,总共10类。
# 读入数据
data_time = pd.read_csv(r"C:\Users\ISTLN\PycharmProjects\pythonProject1\CWRU_data\feature_time_48k_2048_load_1.csv")
fault_typ