Remove hard coded spark mode in xgboost examples (#5579)

intel-analytics · Aug 30, 2022 · c957850 · c957850
1 parent 37a4ded
commit c957850
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/python/dllib/examples/nnframes/xgboost/xgboost_classifier.py b/python/dllib/examples/nnframes/xgboost/xgboost_classifier.py
@@ -37,7 +37,7 @@
 
 
 def process(filepath, demo):
-    sparkConf = init_spark_conf().setMaster("local[1]").setAppName("testXGBClassifier")
+    sparkConf = init_spark_conf().setAppName("testXGBClassifier")
     sc = init_nncontext(sparkConf)
     sqlContext = SQLContext(sc)
     if demo:

diff --git a/python/dllib/examples/nnframes/xgboost/xgboost_example.py b/python/dllib/examples/nnframes/xgboost/xgboost_example.py
@@ -43,7 +43,7 @@ def Processdata(filepath, demo):
     :param filepath:
     :return: assembledf:
     '''
-    sparkConf = init_spark_conf().setMaster("local[1]").setAppName("testNNClassifer")
+    sparkConf = init_spark_conf().setAppName("testNNClassifer")
     sc = init_nncontext(sparkConf)
     sqlContext = SQLContext(sc)
     if demo:

diff --git a/python/dllib/examples/nnframes/xgboost/xgboost_regressor.py b/python/dllib/examples/nnframes/xgboost/xgboost_regressor.py
@@ -42,7 +42,7 @@ def Processdata(filepath, demo):
     :param filepath:
     :return: assembledf:
     '''
-    sparkConf = init_spark_conf().setMaster("local[1]").setAppName("testNNClassifer")
+    sparkConf = init_spark_conf().setAppName("testNNClassifer")
     sc = init_nncontext(sparkConf)
     sqlContext = SQLContext(sc)
     if demo: