测试知识图谱的数据,测试3天

tensnsorflow/es/feature.py

@@ -54,7 +54,7 @@ def get_data():
     print("esmm data ok")
 
     # print(df["method","min","max","treatment_time","maintain_time","recover_time"].head(6))
-    print(df.count())
+    # print(df.count())
 
     service_id = tuple(df["service_id"].unique())
     db = pymysql.connect(host='rdsfewzdmf0jfjp9un8xj.mysql.rds.aliyuncs.com', port=3306, user='work',

tensnsorflow/es/to_tfrecord.py

@@ -29,7 +29,8 @@ def gen_tfrecords(in_file):
 
     for i in range(df.shape[0]):
         feats = ["ucity_id", "ccity_name", "device_type", "manufacturer",
-                "channel", "top", "time", "stat_date","hospital_id"]
+                "channel", "top", "time", "stat_date","hospital_id",
+                 "method", "min", "max", "treatment_time", "maintain_time", "recover_time"]
         id = np.array([])
         for j in feats:
             id = np.append(id,df[j][i])