将 Model.Predict 的输出转换为整数答案

【问题标题】：Convert Output of Model.Predict to integer将 Model.Predict 的输出转换为整数
【发布时间】：2021-07-11 05:38:41
【问题描述】：

devs 我正在使用一个简单的卷积神经网络进行二进制分类，一切运行良好，我正在寻找一种方法将我的模型预测的输出转换为整数（0 或 1）。

在最后一步我有这个说明：

res_f = my_model.predict(test_img)

当我打印 res_f 的值时，结果如下所示：[[0. 1.]] 我的问题是将 res_f 转换为整数？

【问题讨论】：

【解决方案1】：

模型的输出采用 One Hot Encoded 格式。您可以使用argmax 从 One Hot Encoded Array 中获取等效整数。

代码片段：

import numpy as np

res_f = my_model.predict(test_img)
result = np.argmax(res_f)

输出：

【讨论】：