修改字典生成方式

tensnsorflow/feature_engineering.py

@@ -56,7 +56,8 @@ def get_data():
         unique_values.extend(list(df[i].unique()))
     print(df.head(2))
 
-    value_map = {v: k for k, v in enumerate(unique_values)}
+    temp = list(range(1,len(unique_values)+1))
+    value_map = dict(zip(unique_values,temp))
 
     df = df.drop("device_id", axis=1)
     train = df[df["stat_date"] != validate_date+"stat_date"]