Skip to content

Commit 4617f04

Browse files
committed
add python example to loadLibSVMFile and fix Java example
1 parent d6509c2 commit 4617f04

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

docs/mllib-basics.md

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ import org.apache.spark.mllib.regression.LabeledPoint
193193
import org.apache.spark.mllib.util.MLUtils
194194
import org.apache.spark.rdd.RDD
195195

196-
val training: RDD[LabeledPoint] = MLUtils.loadLibSVMFile(sc, "mllib/data/sample_libsvm_data.txt")
196+
val examples: RDD[LabeledPoint] = MLUtils.loadLibSVMFile(sc, "mllib/data/sample_libsvm_data.txt")
197197
{% endhighlight %}
198198
</div>
199199

@@ -204,9 +204,21 @@ examples stored in LIBSVM format.
204204
{% highlight java %}
205205
import org.apache.spark.mllib.regression.LabeledPoint;
206206
import org.apache.spark.mllib.util.MLUtils;
207-
import org.apache.spark.rdd.RDDimport;
207+
import org.apache.spark.api.java.JavaRDD;
208+
209+
JavaRDD<LabeledPoint> examples =
210+
MLUtils.loadLibSVMFile(jsc.sc(), "mllib/data/sample_libsvm_data.txt").toJavaRDD();
211+
{% endhighlight %}
212+
</div>
213+
214+
<div data-lang="python" markdown="1">
215+
[`MLUtils.loadLibSVMFile`](api/python/pyspark.mllib.util.MLUtils-class.html) reads training
216+
examples stored in LIBSVM format.
217+
218+
{% highlight python %}
219+
from pyspark.mllib.util import MLUtils
208220

209-
RDD<LabeledPoint> training = MLUtils.loadLibSVMFile(jsc, "mllib/data/sample_libsvm_data.txt");
221+
examples = MLUtils.loadLibSVMFile(sc, "mllib/data/sample_libsvm_data.txt")
210222
{% endhighlight %}
211223
</div>
212224
</div>

0 commit comments

Comments
 (0)