add hospital and doctor

tensnsorflow/ffm.py

@@ -137,12 +137,14 @@ class multiFFMFormatPandas:
 
 
 def get_data():
-    db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='jerry_test')
-    sql = "select max(stat_date) from esmm_train_data"
-    validate_date = con_sql(db, sql)[0].values.tolist()[0]
+    # db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='jerry_test')
+    # sql = "select max(stat_date) from esmm_train_data"
+    # validate_date = con_sql(db, sql)[0].values.tolist()[0]
+    validate_date = "2018-12-19"
     print("validate_date:" + validate_date)
-    temp = datetime.datetime.strptime(validate_date, "%Y-%m-%d")
-    start = (temp - datetime.timedelta(days=30)).strftime("%Y-%m-%d")
+    # temp = datetime.datetime.strptime(validate_date, "%Y-%m-%d")
+    # start = (temp - datetime.timedelta(days=30)).strftime("%Y-%m-%d")
+    start = "2018-11-19"
     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," \
@@ -197,8 +199,8 @@ def transform(a,validate_date):
     test = test.drop("stat_date",axis=1)
     # print("train shape")
     # print(train.shape)
-    train.to_csv(path + "tr_all.csv", sep="\t", index=False)
-    test.to_csv(path + "va_all.csv", sep="\t", index=False)
+    train.to_csv(path + "tr.csv", sep="\t", index=False)
+    test.to_csv(path + "va.csv", sep="\t", index=False)
 
     return model