From 621b50528dc0741e622d468834128ee1b492837f Mon Sep 17 00:00:00 2001 From: gouravgupta92 Date: Mon, 6 Apr 2015 13:11:55 +0530 Subject: [PATCH] Update HashingTF.scala transform function should return SparseVector instead of Vector. --- .../main/scala/org/apache/spark/mllib/feature/HashingTF.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mllib/src/main/scala/org/apache/spark/mllib/feature/HashingTF.scala b/mllib/src/main/scala/org/apache/spark/mllib/feature/HashingTF.scala index c53475818395f..7e2c9241430f9 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/feature/HashingTF.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/feature/HashingTF.scala @@ -47,7 +47,7 @@ class HashingTF(val numFeatures: Int) extends Serializable { /** * Transforms the input document into a sparse term frequency vector. */ - def transform(document: Iterable[_]): Vector = { + def transform(document: Iterable[_]): SparseVector = { val termFrequencies = mutable.HashMap.empty[Int, Double] document.foreach { term => val i = indexOf(term)