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

Question

I finished training a model with Keras, but I have a problem that I can't get the top K results. I know there is a function called tf.nn.top_k in tensorflow, but how can I use it in Keras?

Answer 1

You can try with the below code.

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