theodotus commited on
Commit
7b8d81e
·
1 Parent(s): 30739e9

Added select_best

Browse files
Files changed (1) hide show
  1. utils.py +19 -1
utils.py CHANGED
@@ -64,4 +64,22 @@ class Test:
64
  answer = question["responces"][answer_idx]
65
  for name in answer["name"]:
66
  name_count[name] += 1
67
- return name_count
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
64
  answer = question["responces"][answer_idx]
65
  for name in answer["name"]:
66
  name_count[name] += 1
67
+ return name_count
68
+
69
+ def select_best(self, answer_ids):
70
+ total_name_count = self.total_name_count()
71
+ current_name_count = self.current_name_count(answer_ids)
72
+
73
+ name_percent = {}
74
+
75
+ for name in self.names:
76
+ percent = current_name_count[name] / total_name_count[name]
77
+ name_percent[name] = percent
78
+
79
+ best_name = self.dict_max(name_percent)
80
+ return best_name
81
+
82
+ @staticmethod
83
+ def dict_max(dict):
84
+ key = max(dict, key=dict.get)
85
+ return key