修改测试文件

021b2dbc · 张彦钊 · 38d63dc0 · 021b2dbc
Commit 021b2dbc authored May 05, 2019 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

multi.py tensnsorflow/multi.py +3 -3

No files found.
--- a/tensnsorflow/multi.py
+++ b/tensnsorflow/multi.py
@@ -227,9 +227,9 @@ if __name__ == '__main__':
        .set("spark.driver.maxResultSize", "8g")

    spark = SparkSession.builder.config(conf=sparkConf).enableHiveSupport().getOrCreate()
-    # ti = pti.TiContext(spark)
-    # ti.tidbMapDatabase("jerry_test")
-    # spark.sparkContext.setLogLevel("WARN")
+    ti = pti.TiContext(spark)
+    ti.tidbMapDatabase("jerry_test")
+    spark.sparkContext.setLogLevel("WARN")
    # validate_date, value_map, app_list_map, leve2_map, leve3_map = feature_engineer()
    # get_predict(validate_date, value_map, app_list_map, leve2_map, leve3_map)