调试

linucb/core/Linucb.py

@@ -14,6 +14,7 @@ import json
 import pickle
 import pymysql
 import random
+import time
 
 
 redis_url = "redis://:ReDis!GmTx*0aN9@172.16.40.173:6379"
@@ -91,10 +92,10 @@ class LinUCB:
                     tag_id = tag_list[top_score_index]
                     top_tag_dict[tag_id] = top_score
                     top_tag_set.add(tag_id)
-                    if len(top_tag_dict) >= 1000:
+                    if len(top_tag_dict) >= 20:
                         break
 
-                if len(top_tag_dict) >= 1000:
+                if len(top_tag_dict) >= 20:
                     break
 
             return (top_tag_dict,top_tag_set)
@@ -225,6 +226,8 @@ if __name__ == "__main__":
 
     test_user_feature = [1, 2, 1, 1, 3, 1]
 
+    begin=time.time()
     all_diary_content_redis_dict = redis_client.hgetall("strategy:linucb:content_type:diary")
     top_tag_dict, top_tag_set = LinUCB.linucb_recommend_tag(all_diary_content_redis_dict,test_user_feature,list(all_diary_content_redis_dict.keys()))
+    print(time.time()-begin)
     print(top_tag_dict)
\ No newline at end of file