add user, item to redis

get_ad_out_sample_v2_item.py

@@ -218,6 +218,10 @@ from candidate_item
     lr_model = LrModel('model/ad_out_v2_model_v1.day.json')
     item_h_dict = {}
     k_col = 'i_id'
+    dt = datetime
+    model_key = 'test_lr_v1'
+    key_name = f"{config_.KEY_NAME_PREFIX_AD_OUT_MODEL_SCORE_ITEM}{model_key}:{dt}"
+    print(key_name)
     for index, row in tqdm(data.iterrows()):
         k = row['i_id']
         item_features = get_item_features(row)
@@ -225,10 +229,6 @@ from candidate_item
         item_h_dict[k] = item_h
         # print(item_features)
         # print(item_h)
-    dt = datetime
-    model_key = 'test_lr_v1'
-    key_name = f"{config_.KEY_NAME_PREFIX_AD_OUT_MODEL_SCORE_ITEM}{model_key}:{dt}"
-    print(key_name)
     redis_helper.add_data_with_zset(key_name=key_name, data=item_h_dict, expire_time=2 * 24 * 3600)
     with open('test_item.json', 'w') as fout:
         json.dump(item_h_dict, fout, indent=2, ensure_ascii=False, sort_keys=True)

get_ad_out_sample_v2_user.py

@@ -218,6 +218,10 @@ from candidate_user
     lr_model = LrModel('model/ad_out_v2_model_v1.day.json')
     item_h_dict = {}
     k_col = 'u_id'
+    dt = datetime
+    model_key = 'test_lr_v1'
+    key_name = f"{config_.KEY_NAME_PREFIX_AD_OUT_MODEL_SCORE_USER}{model_key}:{dt}"
+    print(key_name)
     for index, row in tqdm(data.iterrows()):
         k = row['u_id']
         item_features = get_item_features(row)
@@ -225,10 +229,6 @@ from candidate_user
         item_h_dict[k] = item_h
         # print(item_features)
         # print(item_h)
-    dt = datetime
-    model_key = 'test_lr_v1'
-    key_name = f"{config_.KEY_NAME_PREFIX_AD_OUT_MODEL_SCORE_USER}{model_key}:{dt}"
-    print(key_name)
     redis_helper.add_data_with_zset(key_name=key_name, data=item_h_dict, expire_time=2 * 24 * 3600)
     with open('test_user.json', 'w') as fout:
         json.dump(item_h_dict, fout, indent=2, ensure_ascii=False, sort_keys=True)