keras top 1 accuracy name switch (intel-analytics#2613)

Le-Zheng · Jul 24, 2020 · 45e3c30 · 45e3c30
1 parent a750024
commit 45e3c30
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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',