aboutsummaryrefslogtreecommitdiff
path: root/examples/src/main/python/ml/naive_bayes_example.py
diff options
context:
space:
mode:
Diffstat (limited to 'examples/src/main/python/ml/naive_bayes_example.py')
-rw-r--r--examples/src/main/python/ml/naive_bayes_example.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/examples/src/main/python/ml/naive_bayes_example.py b/examples/src/main/python/ml/naive_bayes_example.py
index e37035542c..89255a2bae 100644
--- a/examples/src/main/python/ml/naive_bayes_example.py
+++ b/examples/src/main/python/ml/naive_bayes_example.py
@@ -24,7 +24,10 @@ from pyspark.ml.evaluation import MulticlassClassificationEvaluator
from pyspark.sql import SparkSession
if __name__ == "__main__":
- spark = SparkSession.builder.appName("naive_bayes_example").getOrCreate()
+ spark = SparkSession\
+ .builder\
+ .appName("naive_bayes_example")\
+ .getOrCreate()
# $example on$
# Load training data