我有一個訓練有素的模型，我正在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（即 a pandas DataFrame）中的行，以便我可以輕松表示輸入到輸出？
1 回答
小唯快跑啊
TA貢獻1863條經驗 獲得超2個贊
默認情況下，model.predict(X)和X本身已經處于相同的順序（的第一個值prediction對應于X的第一行等）。
為了便于可視化，您可以嘗試類似X['predicitons'] = prediction但會向X.
添加回答
舉報
0/150
提交
取消