将 Model.Predict 的输出转换为整数
Convert Output of Model.Predict to integer
devs 我正在使用一个简单的卷积神经网络进行二元分类，一切都运行良好，我正在寻找一种方法将我的模型预测的输出转换为整数（0 或 1）。
在最后一步我有这条指令：
res_f = my_model.predict(test_img)
当我打印 res_f 的值时，结果看起来像：[[0. 1.]]
我的问题是关于将 res_f 转换为整数 ?
模型的输出是 One Hot Encoded 格式。
您可以使用 argmax 从一个热编码数组中获取等效的整数。
代码片段：
import numpy as np
res_f = my_model.predict(test_img)
result = np.argmax(res_f)
输出：
1
devs 我正在使用一个简单的卷积神经网络进行二元分类，一切都运行良好，我正在寻找一种方法将我的模型预测的输出转换为整数（0 或 1）。
在最后一步我有这条指令：
res_f = my_model.predict(test_img)
当我打印 res_f 的值时，结果看起来像：[[0. 1.]] 我的问题是关于将 res_f 转换为整数 ?
模型的输出是 One Hot Encoded 格式。
您可以使用 argmax 从一个热编码数组中获取等效的整数。
代码片段：
import numpy as np
res_f = my_model.predict(test_img)
result = np.argmax(res_f)
输出：
1