modify top_tag_list_len

linucb/views/linucb.py

@@ -59,8 +59,9 @@ class LinUCB:
             AaI_tmp = np.array(Aa_list)
             theta_tmp = np.array(theta_list)
 
-            top_tag_list_len = int(len(tag_list)/3)
-            top_np_ind = np.argpartition(np.dot(xaT, theta_tmp) + cls.alpha * np.sqrt(np.dot(np.dot(xaT, AaI_tmp), xa)), -top_tag_list_len)[-top_tag_list_len:]
+            np_array = np.dot(xaT, theta_tmp) + cls.alpha * np.sqrt(np.dot(np.dot(xaT, AaI_tmp), xa))
+            top_tag_list_len = int(np_array.size/2)
+            top_np_ind = np.argpartition(np_array, -top_tag_list_len)[-top_tag_list_len:]
 
             top_tag_list = list()
             top_np_list = top_np_ind.tolist()