调试

tools/generate_content_user_feature.py

@@ -164,6 +164,36 @@ class Generate_Feature_Info(object):
             logging.error("catch exception,err_msg:%s" % traceback.format_exc())
 
 
+    @classmethod
+    def get_user_feature_by_device_id(cls,device_id):
+        try:
+            redis_name_user_linucb_feature = cls.redis_name_user_linucb_feature_prefix
+            user_feature_redis_data = redis_client.hget(redis_name_user_linucb_feature,device_id)
+
+            user_feature_list = list()
+            if user_feature_redis_data:
+                user_feature_list = [float(json.loads(user_feature_redis_data)[0])]
+
+            return user_feature_list
+        except:
+            logging.error("catch exception,err_msg:%s" % traceback.format_exc())
+            return list()
+
+    @classmethod
+    def get_content_feature(cls,car_id,content_type="diary"):
+        try:
+            redis_name_content_linucb_feature = cls.redis_name_content_linucb_feature_prefix + content_type
+            user_feature_list = list()
+
+            content_feature_redis_data = redis_client.hget(redis_name_content_linucb_feature, car_id)
+            if content_feature_redis_data:
+                content_feature_redis_list = json.loads(content_feature_redis_data)
+                user_feature_list = [float(item) for item in content_feature_redis_list]
+
+            return user_feature_list
+        except:
+            logging.error("catch exception,err_msg:%s" % traceback.format_exc())
+            return list()
 
 
 if __name__ == "__main__":
@@ -172,10 +202,26 @@ if __name__ == "__main__":
 
     label_encoder = Generate_Feature_Info.get_tagv3_label_encode(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)
+    # 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)
+
+
+    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():
+        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_fd)
+
+        print(user_feature_list+content_feature_list)
 
+    diary_fd.close()
     # test_val_list = ["切开双眼皮"]
     #
     # label_results_list = label_encoder.transform(test_val_list)