From 1f8514026d21459f908fba3918631baec80d9906 Mon Sep 17 00:00:00 2001 From: Song Jiaming Date: Fri, 24 Jul 2020 14:05:42 +0800 Subject: [PATCH] keras top 1 accuracy name switch (#2613) --- python/dllib/src/bigdl/dllib/keras/engine/topology.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/dllib/src/bigdl/dllib/keras/engine/topology.py b/python/dllib/src/bigdl/dllib/keras/engine/topology.py index d20fbf7137b..e13fbf62e22 100644 --- a/python/dllib/src/bigdl/dllib/keras/engine/topology.py +++ b/python/dllib/src/bigdl/dllib/keras/engine/topology.py @@ -118,6 +118,8 @@ def get_validation_summary(self, tag=None): # Arguments tag: The string variable represents the scalar wanted """ + if tag == 'Accuracy': + tag = 'Top1Accuracy' validation_set = set(('AUC', 'Accuracy', 'BinaryAccuracy', 'CategoricalAccuracy', 'HitRatio', 'Loss', 'MAE', 'NDCG', 'SparseCategoricalAccuracy', 'TFValidationMethod', 'Top1Accuracy',