Fix early stopping metric check

9214db1d · guptapriya · guptapriya · 8f6e2547 · 9214db1d
Commit 9214db1d authored Jun 11, 2019 by guptapriya Committed by guptapriya Jun 12, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/recommendation/ncf_keras_main.py official/recommendation/ncf_keras_main.py +1 -1

No files found.
--- a/official/recommendation/ncf_keras_main.py
+++ b/official/recommendation/ncf_keras_main.py
@@ -294,7 +294,7 @@ def run_ncf(_):

  if FLAGS.early_stopping:
    early_stopping_callback = CustomEarlyStopping(
-        "val_metric_fn", desired_value=FLAGS.hr_threshold)
+        "val_HR_METRIC", desired_value=FLAGS.hr_threshold)
    callbacks.append(early_stopping_callback)

  strategy = distribution_utils.get_distribution_strategy(