diff options
| -rw-r--r-- | docs/ml-guide.md | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/docs/ml-guide.md b/docs/ml-guide.md index 0427ac6695..fd3a6167bc 100644 --- a/docs/ml-guide.md +++ b/docs/ml-guide.md @@ -475,7 +475,7 @@ val test = sqlContext.createDataFrame(Seq( )).toDF("id", "text") // Make predictions on test documents. -model.transform(test.toDF) +model.transform(test) .select("id", "text", "probability", "prediction") .collect() .foreach { case Row(id: Long, text: String, prob: Vector, prediction: Double) => |