将Keras预测保存到原始数据帧

Question

在Keras模型中，我的数据集以行索引的方式隔离。我的训练，验证和测试集与主数据框（即 Data_Main ）分开的位置如下：

第 startrownumber_train：endrownumber_train 行中的TrainSet，第 startrownumber_validation：endrownumber_validation 行中的ValidationSet以及类似的， startrownumber_test：endrownumber_test

行

TrainSet= Data_Main.loc[startrownumber_train:endrownumber_train,['S','MidPoint_l1','bssigma','rf','yield','T']]
ValidationSet= Data_Main.loc[startrownumber_validation:endrownumber_validation,['S', 'MidPoint_l1', 'bssigma', 'rf', 'yield', 'T']]
TestSet= Data_Main.loc[startrownumber_test:endrownumber_test,['S', 'MidPoint_l1', 'bssigma', 'rf', 'yield', 'T']]

现在让我们假设 startrownumber_test：endrownumber_test 是否来自 100：200 行，该行取自 Data_Main 数据框，然后将TestSet传递给Keras中的 model.predict ，如下所示：

model.predict(TestSet)

然后，如何将上述步骤中的预测输出存储到列中，即 Data_Main 中来自 100：200 行的预测 >数据框？

我尝试做：

Data_Main["Predictions"][startrownumber_test:endrownumber_test]=model.predict(TestSet)

但是给我一个错误。