调试

05eb158a · 段英荣 · f968b8ac · 05eb158a
Commit 05eb158a authored Apr 21, 2020 by 段英荣
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

Linucb.py linucb/core/Linucb.py +1 -2

No files found.
--- a/linucb/core/Linucb.py
+++ b/linucb/core/Linucb.py
@@ -50,9 +50,8 @@ class LinUCB:
                tag_dict = pickle.loads(redis_linucb_tag_data_dict[tag_id])
                Aa_list.append(tag_dict["Aa"])
                theta_list.append(tag_dict["theta"])
-                print(tag_dict["Aa"].shape)

-            xaT = np.array([user_features_list])
+            xaT = np.array(user_features_list)
            xa = np.transpose(xaT)

            art_max = -1