diff --git a/examples/src/main/python/ml/dataframe_example.py b/examples/src/main/python/ml/dataframe_example.py
index d62cf2338a..cabc3de68f 100644
--- a/examples/src/main/python/ml/dataframe_example.py
+++ b/examples/src/main/python/ml/dataframe_example.py
@@ -17,7 +17,7 @@
"""
An example of how to use DataFrame for ML. Run with::
- bin/spark-submit examples/src/main/python/ml/dataframe_example.py
+ bin/spark-submit examples/src/main/python/ml/dataframe_example.py
"""
from __future__ import print_function
@@ -35,18 +35,18 @@ if __name__ == "__main__":
print("Usage: dataframe_example.py ", file=sys.stderr)
sys.exit(-1)
elif len(sys.argv) == 2:
- input = sys.argv[1]
+ input_path = sys.argv[1]
else:
- input = "data/mllib/sample_libsvm_data.txt"
+ input_path = "data/mllib/sample_libsvm_data.txt"
spark = SparkSession \
.builder \
.appName("DataFrameExample") \
.getOrCreate()
- # Load input data
- print("Loading LIBSVM file with UDT from " + input + ".")
- df = spark.read.format("libsvm").load(input).cache()
+ # Load an input file
+ print("Loading LIBSVM file with UDT from " + input_path + ".")
+ df = spark.read.format("libsvm").load(input_path).cache()
print("Schema from LIBSVM:")
df.printSchema()
print("Loaded training data as a DataFrame with " +