Keras设置Top5精度

 

import keras

def acc_top5(y_true, y_pred):
    return keras.metrics.top_k_categorical_accuracy(y_true, y_pred, k=5)

model.compile(loss=loss, optimizer=optimizer, metrics=['accuracy', acc_top5])

如果想设置其他的精度，修改top_k_categorical_accuracy(y_true, y_pred, k=5)里面的k值即可，我这里是设置的top5精度。