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',