diff options
Diffstat (limited to 'examples/src/main/python/kmeans.py')
-rwxr-xr-x | examples/src/main/python/kmeans.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/examples/src/main/python/kmeans.py b/examples/src/main/python/kmeans.py index e3596488fa..fc16586c28 100755 --- a/examples/src/main/python/kmeans.py +++ b/examples/src/main/python/kmeans.py @@ -45,14 +45,14 @@ def closestPoint(p, centers): if __name__ == "__main__": - if len(sys.argv) < 5: - print >> sys.stderr, "Usage: kmeans <master> <file> <k> <convergeDist>" + if len(sys.argv) != 4: + print >> sys.stderr, "Usage: kmeans <file> <k> <convergeDist>" exit(-1) - sc = SparkContext(sys.argv[1], "PythonKMeans") - lines = sc.textFile(sys.argv[2]) + sc = SparkContext(appName="PythonKMeans") + lines = sc.textFile(sys.argv[1]) data = lines.map(parseVector).cache() - K = int(sys.argv[3]) - convergeDist = float(sys.argv[4]) + K = int(sys.argv[2]) + convergeDist = float(sys.argv[3]) kPoints = data.takeSample(False, K, 1) tempDist = 1.0 |