Unverified Commit d1f87352 authored by penolove's avatar penolove Committed by GitHub
Browse files

[python] add average precision into higher_better eval (#3649)

parent b69364e9
......@@ -3292,7 +3292,7 @@ class Booster:
self.__name_inner_eval = \
[string_buffers[i].value.decode('utf-8') for i in range(self.__num_inner_eval)]
self.__higher_better_inner_eval = \
[name.startswith(('auc', 'ndcg@', 'map@')) for name in self.__name_inner_eval]
[name.startswith(('auc', 'ndcg@', 'map@', 'average_precision')) for name in self.__name_inner_eval]
def attr(self, key):
"""Get attribute string from the Booster.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment