modify linucb

linucb/views/linucb.py

@@ -102,18 +102,21 @@ class LinUCB:
                 redis_key = redis_prefix + str(device_id)
                 ori_redis_tag_dict = redis_cli.hget(redis_key, tag_id)
 
-                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
-
-                user_tag_dict = {
-                    "Aa": new_Aa_matrix,
-                    "ba": new_ba_matrix,
-                    "AaI": new_AaI_matrix,
-                    "theta": np.dot(new_AaI_matrix, new_ba_matrix)
-                }
-
-                redis_cli.hset(redis_key, tag_id, user_tag_dict)
+                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
+
+                    user_tag_dict = {
+                        "Aa": new_Aa_matrix,
+                        "ba": new_ba_matrix,
+                        "AaI": new_AaI_matrix,
+                        "theta": np.dot(new_AaI_matrix, new_ba_matrix)
+                    }
+
+                    redis_cli.hset(redis_key, tag_id, user_tag_dict)
             return True
         except:
             logging.error("catch exception,err_msg:%s" % traceback.format_exc())