gpt4 book ai didi

tensorflow - Keras 前 5 名预测

转载 作者:行者123 更新时间:2023-12-02 02:55:13 26 4
gpt4 key购买 nike

如何从 model.predict() 中检索前 5 个预测在凯拉斯?它只给出 1 个预测。有什么办法吗?我不希望它作为评估指标。我只需要前 5 个预测。

最佳答案

如果您想从图像分类问题中获得最高预测,
您将收到一个热门代码预测。

class_prob = [0.98,0.50,0.60,0.90,0.87,0.79,0.87]
top_values_index = sorted(range(len(class_prob)), key=lambda i: class_prob[i])[-the_top_values_you_want_to_extract:]

您现在拥有所有五个最高值的索引。您现在可以循环遍历索引并获取类名。

只提取 top_values_without_index
top_values= [class_prob[i] for i in np.argsort(class_prob)[-5:]]

关于tensorflow - Keras 前 5 名预测,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49827838/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com