如何使用 keras.prediction 将测试数据与预测数据对齐？

Question

我有一个训练有素的模型，我正在通过 keras 运行预测：

model = pets.get_model(input_size=input_units)
model.compile(loss='categorical_crossentropy',
              optimizer='adam', metrics=['accuracy'])
model.load_weights('models/2019-03-01-02-03-53.h5')

prediction = model.predict(X)

这给了我一个看起来像[0.323 0.43 .099] 的列表等等。如何将其映射到我的X（这是一个pandasDataFrame）中的行，以便我可以轻松地表示输入到输出？

Answer 1

默认情况下，model.predict(X) 和 X 本身已经处于相同的顺序（prediction 的第一个值对应于X 的第一行，依此类推）。

为了便于可视化，您可以尝试X['predicitons'] = prediction 之类的方法，但这会在X 中添加一列。