如何使用Keras进行预测后将测试数据与预测数据对齐？

Question

如何使用Keras进行预测后将测试数据与预测数据对齐？

4

我有一个训练好的模型，现在正在使用 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（它是一个pandas DataFrame），以便我可以轻松地表示输入和输出？

- Shamoon

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Victor Valente · Accepted Answer

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

为了方便可视化，您可以尝试类似这样的操作：X ['predicitons'] = prediction，但这会向X添加一列。