重构lr_v1

get_ad_out_sample_v2_item.py

@@ -16,7 +16,7 @@ from feature import get_item_features
 from lr_model import LrModel
 from utils import exe_sql
 
-if __name__=="__main__":
+if __name__ == "__main__":
     project = 'loghubods'
     datetime = sys.argv[1]
     sql = """
@@ -212,8 +212,8 @@ from candidate_item
     print(sql)
     data = exe_sql(project, sql)
     print('sql done')
-    #data.to_csv('./data/ad_out_sample_v2_item.{datetime}'.format(datetime=datetime), sep='\t')
-    #data = pd.read_csv('./data/ad_out_sample_v2_item.{datetime}'.format(datetime=datetime), sep='\t', dtype=str)
+    # data.to_csv('./data/ad_out_sample_v2_item.{datetime}'.format(datetime=datetime), sep='\t')
+    # data = pd.read_csv('./data/ad_out_sample_v2_item.{datetime}'.format(datetime=datetime), sep='\t', dtype=str)
     data.fillna('', inplace=True)
     lr_model = LrModel('model/ad_out_v2_model_v1.day.json')
     item_h_dict = {}
@@ -223,11 +223,11 @@ from candidate_item
         item_features = get_item_features(row)
         item_h = lr_model.predict_h(item_features)
         item_h_dict[k] = item_h
-        #print(item_features)
-        #print(item_h)
+        # print(item_features)
+        # print(item_h)
     dt = datetime
-    data_key = 'test_lr_v1'
-    key_name = f"{config_.KEY_NAME_PREFIX_AD_GROUP}{data_key}:{dt}"
+    model_key = 'test_lr_v1'
+    key_name = f"{config_.KEY_NAME_PREFIX_AD_VIDEO}{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: