change df to tfrecord

tensnsorflow/feature_engineering.py

@@ -25,7 +25,7 @@ def get_data():
     start = (temp - datetime.timedelta(days=30)).strftime("%Y-%m-%d")
     print(start)
     db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='jerry_test')
-    sql = "select e.y,e.z,e.stat_date,e.ucity_id,e.clevel1_id,e.ccity_name," \
+    sql = "select e.y,e.z,e.stat_date,e.ucit_y_id,e.clevel1_id,e.ccity_name," \
           "u.device_type,u.manufacturer,u.channel,c.top,df.level2_ids,e.device_id,cut.time " \
           "from esmm_train_data e left join user_feature u on e.device_id = u.device_id " \
           "left join cid_type_top c on e.device_id = c.device_id " \

tensnsorflow/test.py

@@ -43,7 +43,7 @@ def gen_tfrecords(in_file):
             "stat_date": tf.train.Feature(int64_list=tf.train.Int64List(value=[df["stat_date"][i]]))
         })
 
-        example = tf.train.Example(features = tf.train.Features(feature = features))
+        example = tf.train.Example(features = features)
         serialized = example.SerializeToString()
         tfrecord_out.write(serialized)
     tfrecord_out.close()