增加特征

d0ed7c24 · 张彦钊 · a3dd14f0 · d0ed7c24
Commit d0ed7c24 authored May 13, 2019 by 张彦钊
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

feature.py eda/esmm/Model_pipline/feature.py +2 -2

No files found.
--- a/eda/esmm/Model_pipline/feature.py
+++ b/eda/esmm/Model_pipline/feature.py
@@ -120,8 +120,8 @@ def get_data():

    df = df.drop("device_id", axis=1)
    # TODO 上线后把最近一天的数据集放进训练集，这样用户的正、负反馈能及时获取
-    # train = df
-    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",