asdf

linucb/core/Linucb.py

@@ -210,23 +210,36 @@ class LinUCB:
 
 
 if __name__ == "__main__":
+    # user_features = [
+    #     [1, 2, 1, 1, 3, 1],
+    #     [1, 4, 3, 1, 3, 1],
+    #     [3, 2, 1, 5, 5, 1],
+    #     [1, 2, 4, 2, 3, 2],
+    #     [1, 5, 7, 1, 4, 4],
+    #     [3, 4, 1, 1, 3, 1],
+    #     [5, 2, 1, 6, 3, 1],
+    #     [1, 2, 3, 2, 3, 5],
+    #     [1, 2, 1, 1, 2, 4],
+    #     [1, 2, 6, 4, 2, 1],
+    # ]
+
     user_features = [
-        [1, 2, 1, 1, 3, 1],
-        [1, 4, 3, 1, 3, 1],
-        [3, 2, 1, 5, 5, 1],
-        [1, 2, 4, 2, 3, 2],
-        [1, 5, 7, 1, 4, 4],
-        [3, 4, 1, 1, 3, 1],
-        [5, 2, 1, 6, 3, 1],
-        [1, 2, 3, 2, 3, 5],
-        [1, 2, 1, 1, 2, 4],
-        [1, 2, 6, 4, 2, 1],
+        ["1", "2", "9", "1", "3", "b"],
+        ["2", "2", "1", "2", "2", "1"],
+        ["3", "7", "1", "1", "3", "1"],
+        ["4", "2", "4", "5", "3", "a"],
+        ["3", "2", "r", "1", "4", "1"],
+        ["5", "5", "2", "3", "3", "1"],
+        ["3", "2", "1", "1", "3", "1"],
+        ["7", "3", "5", "1", "5", "1"],
+        ["8", "2", "1", "9", "3", "c"],
+        ["2", "2", "7", "1", "32", "1"],
     ]
 
-    #LinUCB.init_all_arm_by_card_content(user_features_list=user_features)
+    LinUCB.init_all_arm_by_card_content(user_features_list=user_features)
 
 
-    test_user_feature = [1, 2, 1, 1, 3, 1]
+    test_user_feature = ["1", "2", "9", "1", "3", "b"]
 
     begin=time.time()
     all_diary_content_redis_dict = redis_client.hgetall("strategy:linucb:content_type:diary")