From 7d3149a21f9fbdc416bd09e28305d016651012c6 Mon Sep 17 00:00:00 2001 From: Philip Hyunsu Cho Date: Fri, 23 Nov 2018 12:15:34 -0800 Subject: [PATCH] Add AUC-PR to list of metrics to maximize for early stopping (#3936) --- python-package/xgboost/callback.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python-package/xgboost/callback.py b/python-package/xgboost/callback.py index 1d879a7ae..a4bff8113 100644 --- a/python-package/xgboost/callback.py +++ b/python-package/xgboost/callback.py @@ -188,8 +188,8 @@ def early_stop(stopping_rounds, maximize=False, verbose=True): msg = ("Multiple eval metrics have been passed: " "'{0}' will be used for early stopping.\n\n") rabit.tracker_print(msg.format(env.evaluation_result_list[-1][0])) - maximize_metrics = ('auc', 'map', 'ndcg') - maximize_at_n_metrics = ('auc@', 'map@', 'ndcg@') + maximize_metrics = ('auc', 'aucpr', 'map', 'ndcg') + maximize_at_n_metrics = ('auc@', 'aucpr@' 'map@', 'ndcg@') maximize_score = maximize metric_label = env.evaluation_result_list[-1][0] metric = metric_label.split('-', 1)[-1]