Fixed typo in DecisionTreeModel.scala doc

jkbradley · jkbradley · commit 306120fc9302 · 2014-09-09T20:09:02.000-07:00
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/tree/model/DecisionTreeModel.scala b/mllib/src/main/scala/org/apache/spark/mllib/tree/model/DecisionTreeModel.scala
@@ -46,7 +46,7 @@ class DecisionTreeModel(val topNode: Node, val algo: Algo) extends Serializable
    * Predict values for the given data set using the model trained.
    *
    * @param features RDD representing data points to be predicted
-   * @return RDD[Int] where each entry contains the corresponding prediction
+   * @return RDD of predictions for each of the given data points
    */
   def predict(features: RDD[Vector]): RDD[Double] = {
     features.map(x => predict(x))

Original file line number	Diff line number	Diff line change
`@@ -46,7 +46,7 @@ class DecisionTreeModel(val topNode: Node, val algo: Algo) extends Serializable`
`46`	`46`	`* Predict values for the given data set using the model trained.`
`47`	`47`	`*`
`48`	`48`	`* @param features RDD representing data points to be predicted`
`49`		`- * @return RDD[Int] where each entry contains the corresponding prediction`
	`49`	`+ * @return RDD of predictions for each of the given data points`
`50`	`50`	`*/`
`51`	`51`	`def predict(features: RDD[Vector]): RDD[Double] = {`
`52`	`52`	`features.map(x => predict(x))`