change test file

tensnsorflow/multi.py

@@ -84,7 +84,7 @@ def feature_engineer():
 
 
 def get_predict(date,value_map,app_list_map):
-    sql = "select e.y,e.z,e.label,e.ucity_id,feat.level2_ids from esmm_pre_data e " \
+    sql = "select e.y,e.z,e.label,e.ucity_id,feat.level2_ids,e.device_id,e.cid_id from esmm_pre_data e " \
           "left join diary_feat feat on e.cid_id = feat.diary_id"
 
     features = ["ucity_id"]
@@ -137,7 +137,7 @@ if __name__ == '__main__':
     spark = SparkSession.builder.config(conf=sparkConf).enableHiveSupport().getOrCreate()
     ti = pti.TiContext(spark)
     ti.tidbMapDatabase("jerry_test")
-    ti.tidbMapDatabase("eagle")
+    # ti.tidbMapDatabase("eagle")
     spark.sparkContext.setLogLevel("WARN")
     path = "hdfs:///strategy/esmm/"
     local_path = "/home/gmuser/esmm/"