Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix early stopping behavior with MAPE metric (#7061)

Browse Source
This commit is contained in:
Philip Hyunsu Cho 2021-06-25 12:02:33 -07:00 committed by GitHub
parent 8fa32fdda2
commit dd4db347f3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python-package/xgboost/callback.py
View File

@ -549,7 +549,7 @@ class EarlyStopping(TrainingCallback):
# user to decide. # user to decide.
maximize_metrics = ('auc', 'aucpr', 'map', 'ndcg', 'auc@', maximize_metrics = ('auc', 'aucpr', 'map', 'ndcg', 'auc@',
'aucpr@', 'map@', 'ndcg@') '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 self.maximize = True
else: else:
self.maximize = False self.maximize = False