训练后的Keras top k准确性

时间:2018-09-09 14:59:39

标签: python tensorflow keras

我想在模型训练后获得最高的k精度。我不想为实现最高的top k精度而进行显式优化,而只是想出了训练模型后得出的top k精度是多少。

到目前为止,我所看到的所有示例都要求在编译期间将top_k作为度量提供。

1 个答案:

答案 0 :(得分:0)

即使您想在训练后获得top_k accuracy,您仍然可以使用tf.keras.metrics.top_k_categorical_accuracy。您可以先进行预测,然后使用指标来获取top_k accuracy,如下所示:

predictions = model.predict(x_test)

output = tf.keras.metrics.top_k_categorical_accuracy(one_hot_valued_y_test, predictions, k=5)

with tf.Session() as sess:
    result = sess.run(output)

print(result)