Skip to content
Snippets Groups Projects
Commit 5c202d11 authored by Sun Jin Kim's avatar Sun Jin Kim
Browse files

fix get_n_best_policies()

parent 015087bd
No related branches found
No related tags found
No related merge requests found
Pipeline #273723 passed
......@@ -476,8 +476,8 @@ class AaLearner:
list of best n policies
"""
number_policies = max(number_policies, len(self.history))
number_policies = min(number_policies, len(self.history))
inter_pol = sorted(self.history, key=lambda x: x[1], reverse = True)[:number_policies]
return inter_pol[:number_policies]
return inter_pol
......@@ -251,11 +251,11 @@ class UcbLearner(RsLearner):
temp_history = list(zip(self.policies, temp_avg_accs))
number_policies = max(number_policies, len(temp_history))
number_policies = min(number_policies, len(temp_history))
inter_pol = sorted(temp_history, key=lambda x: x[1], reverse = True)[:number_policies]
return inter_pol[:number_policies]
return inter_pol
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment