add doctor

tensnsorflow/ffm.py

@@ -146,7 +146,7 @@ def get_data():
     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," \
-          "u.device_type,u.manufacturer,u.channel,c.top,cid_time.time,s.hospital_id " \
+          "u.device_type,u.manufacturer,u.channel,c.top,cid_time.time,s.doctor_id " \
           "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 left join cid_time on e.cid_id = cid_time.cid_id " \
           "left join service_hospital s on e.diary_service_id = s.id " \
@@ -196,8 +196,8 @@ def transform(a,validate_date):
     test = test.drop("stat_date",axis=1)
     # print("train shape")
     # print(train.shape)
-    train.to_csv(path + "tr.csv", sep="\t", index=False)
-    test.to_csv(path + "va.csv", sep="\t", index=False)
+    train.to_csv(path + "tr_doctor.csv", sep="\t", index=False)
+    test.to_csv(path + "va_doctor.csv", sep="\t", index=False)
 
     return model