标签： python keras prediction
我已经完成了用Keras训练模型的操作，但是我有一个问题，我无法获得前K个结果。我知道张量流中有一个名为tf.nn.top_k的函数，但是如何在Keras中使用它呢？
tf.nn.top_k
答案 0 :(得分：1)
您可以尝试以下代码。
y_preds = self.model.predict(x) y_preds = np.argsort(y_preds, axis=1)[:,-n:]