Fix early stopping behavior with MAPE metric (#7061)

This commit is contained in:
Philip Hyunsu Cho
2021-06-25 12:02:33 -07:00
committed by GitHub
parent 8fa32fdda2
commit dd4db347f3

View File

@@ -549,7 +549,7 @@ class EarlyStopping(TrainingCallback):
# user to decide.
maximize_metrics = ('auc', 'aucpr', 'map', 'ndcg', 'auc@',
'aucpr@', 'map@', 'ndcg@')
if any(metric.startswith(x) for x in maximize_metrics):
if metric != 'mape' and any(metric.startswith(x) for x in maximize_metrics):
self.maximize = True
else:
self.maximize = False