Add support for keras model.compiled_metrics and model.compiled_loss …

…that are None. PiperOrigin-RevId: 365835530
heejvely · Mar 30, 2021 · 78a01f6 · 78a01f6
1 parent edf0bbc
commit 78a01f6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorflow_model_analysis/evaluators/keras_util.py b/tensorflow_model_analysis/evaluators/keras_util.py
@@ -62,8 +62,8 @@ def metric_computations_using_keras_saved_model(
   # need to call model.evaluate.
   if not model.metrics:
     return []
-  elif (hasattr(model, 'compiled_metrics') and
-        hasattr(model, 'compiled_loss') and
+  elif (hasattr(model, 'compiled_metrics') and model.compiled_metrics and
+        hasattr(model, 'compiled_loss') and model.compiled_loss and
         len(model.compiled_metrics.metrics) + len(model.compiled_loss.metrics)
         == len(model.metrics)):
     output_names = model.output_names if hasattr(model, 'output_names') else []