调试

linucb/core/Linucb.py

@@ -60,7 +60,9 @@ class LinUCB:
             AaI_tmp = np.array(Aa_list)
             theta_tmp = np.array(theta_list)
 
+            begin1 = time.time()
             np_array = np.dot(xaT, theta_tmp) + cls.alpha * np.sqrt(np.dot(np.dot(xaT, AaI_tmp), xa))
+            print(time.time()-begin1)
             # top_tag_list_len = int(np_array.size/2)
             # top_np_ind = np.argpartition(np_array, -top_tag_list_len)[-top_tag_list_len:]
             #
@@ -226,8 +228,8 @@ if __name__ == "__main__":
 
     test_user_feature = [1, 2, 1, 1, 3, 1]
 
-    all_diary_content_redis_dict = redis_client.hgetall("strategy:linucb:content_type:diary")
     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