我用Keras训练完了一个模型，但是我遇到了一个问题，就是无法得到前K的结果。我知道有一个函数叫做 tf.nn.top_k 在tensorflow中使用，但我如何在Keras中使用它？
tf.nn.top_k
你可以用下面的代码试试。
y_preds = self.model.predict(x) y_preds = np.argsort(y_preds, axis=1)[:,-n:]