如何为超过两个类别获取预测标签？

6 浏览
0 Comments

如何为超过两个类别获取预测标签？

我使用TensorFlow后端在Keras中实现了一个图像分类器。对于一个有两个输出类别的数据集，我检查了预测的标签：

if  result[0][0] == 1:
    prediction ='adathodai'
else:
    prediction ='thamarathtai'

完整代码。对于三个类别，我得到[[0. 0. 1.]]。如何以if else的格式检查多于两个类别的预测标签？

0