修改模型训练文件

prepareData.py

@@ -14,7 +14,8 @@ def fetch_data(start_date, end_date):
     # 获取点击表里的数据
     sql = "select cid,device_id,time,stat_date from data_feed_click2 " \
           "where stat_date >= '{0}' and stat_date <= '{1}'".format(start_date, end_date)
-
+    # 因为上面的db已经关了，需要再写一遍
+    db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='jerry_test')
     click = con_sql(db,sql)
     click = click.rename(columns={0: "cid", 1: "device_id", 2: "time_date", 3: "stat_date"})
     print("成功获取点击表里的数据")
@@ -27,7 +28,7 @@ def fetch_data(start_date, end_date):
     sql = "select cid,device_id,time,stat_date from data_feed_exposure2 " \
           "where stat_date >= '{0}' and stat_date <= '{1}'".format(start_date, end_date)
     start = time.time()
-    # 因为上面的db已经关了，索引需要在写一遍
+    # 因为上面的db已经关了，需要再写一遍
     db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='jerry_test')
     exposure = con_sql(db,sql)
     end = time.time()