data_train = np.asarray(data_normal[:1400] + data_inner[:1400] + data_ball[:1400] + data_outer[:1400],dtype = 'float64')
label = np.asarray(label_normal[:1400] + label_inner[:1400] + label_ball[:1400] + label_outer[:1400],dtype = 'int64')
print("处理后样本shape:",data_train.shape)
print("处理后数据类别分布:",Counter(label))
# 保存数据
np.save("train_data/train_data.npy",data_train)
np.save("train_data/label.npy",label)
print("数据保存成功,位置:/train_data/")
【无标题】
于 2022-05-09 10:18:28 首次发布