调试

tools/generate_content_user_feature.py

@@ -9,6 +9,7 @@ import numpy as np
 import sys
 sys.path.append("/home/gmuser/gm_mab/")
 from libs.cache import redis_client
+from linucb.core.Linucb import *
 
 import logging
 import traceback
@@ -210,6 +211,8 @@ if __name__ == "__main__":
     # Generate_Feature_Info.generate_content_feature_to_redis(label_encoder)
 
 
+    linucb_matrix_redis_name = "strategy:linucb:matrix:content_type:diary"
+
     diary_click_file = "/data/log/duan_test/feed_query_data/feed_click_info.txt"
     diary_fd = open(diary_click_file,"r")
 
@@ -227,8 +230,11 @@ if __name__ == "__main__":
         user_feature_list = Generate_Feature_Info.get_user_feature_by_device_id(device_id)
         content_feature_list = Generate_Feature_Info.get_content_feature(diary_id)
 
-        print(user_feature_list+content_feature_list)
+        user_feature_list = user_feature_list+content_feature_list
+        print(user_feature_list)
 
+        LinUCB.update_linucb_info(user_features=user_feature_list,reward=1,content_id=diary_id,
+                                  redis_name_linucb_matrix=linucb_matrix_redis_name,redis_cli=redis_client)
     diary_fd.close()
     # test_val_list = ["切开双眼皮"]
     #