如果不想删除第一列,只需要将`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
```