把最近一天的数据集放进训练集

eda/esmm/Model_pipline/feature.py

@@ -105,7 +105,8 @@ def get_data():
     value_map = dict(zip(unique_values, temp))
 
     df = df.drop("device_id", axis=1)
-    train = df[df["stat_date"] != validate_date + "stat_date"]
+    # 最近一天的数据集放进训练集，这样用户的正、负反馈能及时获取
+    train = df
     test = df[df["stat_date"] == validate_date + "stat_date"]
     for i in ["ucity_id", "ccity_name", "device_type", "manufacturer",
               "channel", "top", "time", "stat_date", "hospital_id",
@@ -113,8 +114,8 @@ def get_data():
         train[i] = train[i].map(value_map)
         test[i] = test[i].map(value_map)
 
-    print("train shape")
-    print(train.shape)
+    # print("train shape")
+    # print(train.shape)
     print("test shape")
     print(test.shape)