modify linucb

linucb/views/linucb.py

@@ -102,6 +102,9 @@ class LinUCB:
                 redis_key = redis_prefix + str(device_id)
                 ori_redis_tag_dict = redis_cli.hget(redis_key, tag_id)
 
+                if not ori_redis_tag_dict:
+                    LinUCB.init_device_id_linucb_info(redis_client, redis_prefix, device_id,[tag_id])
+                else:
                     new_Aa_matrix = ori_redis_tag_dict["Aa"] + np.dot(xa, xaT)
                     new_AaI_matrix = np.linalg.solve(new_Aa_matrix, np.identity(cls.d))
                     new_ba_matrix = ori_redis_tag_dict["ba"] + r*xa