Python 如何使用Keras获得前K预测

我用Keras完成了一个模型的训练，但我有一个问题，我不能得到前K的结果。我知道tensorflow中有一个名为tf.nn.top_k的函数，但是我如何在Keras中使用它呢？

您可以尝试使用下面的代码

y_preds = self.model.predict(x)
y_preds = np.argsort(y_preds, axis=1)[:,-n:]

---

如果问题指定您是为了损失计算、度量还是仅仅为了解释推断而计算topN，则可能的重复项将非常有用。这些是不同的场景。将所有预测存储在一个列表中，并应用sortedlist_name，reverse=True并打印列表的前5个预测