在这个代码的基础上,如果不想删除第一列,要怎么改这个代码def startTrainMult(data,name,config): ''' data: 多维数据 返回训练好的模型 ''' data =

动态 未结 0 108
小小草
小小草 LV1 2023年9月23日 23:40 发表
如果不想删除第一列,只需要将`data = data.iloc[:,1:]`这行代码注释掉即可。修改后的代码如下: ```python def startTrainMult(data,name,config): ''' data: 多维数据 返回训练好的模型 ''' # data = data.iloc[:,1:] # 注释掉这行代码 print(data.columns) yindex = data.columns.get_loc(name) data = np.array(data,dtype='float64') #数据归一化 data, normalize = NormalizeMult(data) data_y = data[:,yindex] data_y = data_y.reshape(data_y.shape[0],1) print(data.shape, data_y.shape) #构造训练数据 trainX1,trainX2, _ = create_dataset(data, config.n_predictions,config.skip) _ , _,trainY = create_dataset(data_y,config.n_predictions,config.skip) print("trainX Y shape is:",trainX1.shape,trainX2.shape,trainY.shape) if len(trainY.shape) == 1: trainY = trainY.reshape(-1,1) # 进行训练 model = trainModel(trainX1, trainX2 , trainY, config) return model,normalize ```
收藏(0)  分享
相关标签: 智能问答
问题没解决?让AI助手帮你作答 AI助手
0个回复
  • 消灭零回复