调试

tools/generate_content_user_feature.py

@@ -77,7 +77,7 @@ class Generate_Feature_Info(object):
         try:
             label_encode_results = label_encoder.transform(tagv3_name_list)
             label_encode_results = np.array([label_encode_results]).T
-            onehot_encoder = OneHotEncoder().fit_transform(label_encode_results)
+            onehot_encoder = OneHotEncoder().fit(label_encode_results)
             return onehot_encoder
         except:
             logging.error("catch exception,err_msg:%s" % traceback.format_exc())
@@ -222,39 +222,40 @@ if __name__ == "__main__":
     label_encoder = Generate_Feature_Info.get_tagv3_label_encode(tagv3_name_list=list(tagv3_name_set))
     onehot_encoder = Generate_Feature_Info.get_tagv3_label_onehot(label_encoder=label_encoder,tagv3_name_list=list(tagv3_name_set))
 
+
     # device_id="868771031984211"
     # Generate_Feature_Info.generate_user_feature_to_redis(device_id,label_encoder)
     # 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")
-    #
-    # for line in diary_fd.readlines():
-    #     line = line.strip()
-    #     line = line.strip("\n")
-    #     line = line.strip("\r")
-    #     line = line.strip(" ")
-    #
-    #     item_list = line.split(",")
-    #     device_id = item_list[3]
-    #     diary_id = item_list[4]
-    #     Generate_Feature_Info.generate_user_feature_to_redis(device_id, label_encoder)
-    #
-    #     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)
-    #
-    #     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 = ["切开双眼皮"]
 
-    label_results_list = label_encoder.transform(test_val_list)
-    label_one_hot_results_list = onehot_encoder.transform(label_results_list)
 
-    print(label_one_hot_results_list)
\ No newline at end of file
+    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")
+
+    for line in diary_fd.readlines():
+        line = line.strip()
+        line = line.strip("\n")
+        line = line.strip("\r")
+        line = line.strip(" ")
+
+        item_list = line.split(",")
+        device_id = item_list[3]
+        diary_id = item_list[4]
+        Generate_Feature_Info.generate_user_feature_to_redis(device_id, label_encoder)
+
+        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)
+
+        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 = ["切开双眼皮"]
+    #
+    # label_results_list = label_encoder.transform(test_val_list)
+    # label_one_hot_results_list = onehot_encoder.transform(label_results_list)
+    #
+    # print(label_one_hot_results_list)
\ No newline at end of file