我有一个经过训练的模型,我通过以下方式使用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)中的行,以便轻松地将输入表示为输出?
发布于 2019-03-01 23:57:52
默认情况下,model.predict(X)和X本身的顺序已经相同( prediction的第一个值对应于X的第一行,依此类推)。
为了便于可视化,您可以尝试类似于X['predicitons'] = prediction的内容,但这会向X添加一列。
https://stackoverflow.com/questions/54948149
复制相似问题