调试

linucb/core/Linucb.py

@@ -51,7 +51,7 @@ class LinUCB:
                 Aa_list.append(tag_dict["Aa"])
                 theta_list.append(tag_dict["theta"])
 
-            xa = np.array(user_features_list)
+            xa = np.array([user_features_list])
             xaT = np.transpose(xa)
 
             art_max = -1
@@ -79,7 +79,7 @@ class LinUCB:
             np_score_list = list()
             np_score_dict = dict()
 
-            import pdb;pdb.set_trace()
+            # import pdb;pdb.set_trace()
             for score_index in range(0,np_array.size):
                 score = np_array.take(score_index)
                 np_score_list.append(score)
@@ -88,7 +88,7 @@ class LinUCB:
                 else:
                     np_score_dict[score].append(score_index)
 
-            import pdb;pdb.set_trace()
+            # import pdb;pdb.set_trace()
             sorted_np_score_list = sorted(np_score_list,reverse=True)
             for top_score in sorted_np_score_list:
                 for top_score_index in np_score_dict[top_score]: