aboutsummaryrefslogtreecommitdiff
path: root/examples/src/main/python/kmeans.py
diff options
context:
space:
mode:
Diffstat (limited to 'examples/src/main/python/kmeans.py')
-rwxr-xr-xexamples/src/main/python/kmeans.py12
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