数字上的Spark IDFModel

时间:2016-09-29 02:32:47

标签: apache-spark-mllib tf-idf

我想在"文件"的数据上执行TF-IDF模型。内容是数字标识符(而不是文本)。所以我不想哈希它们,只需使用数值。有没有简单的方法来生成org.apache.spark.mllib.linalg.VectorUDT?我是否必须编写自己的非散列HashingTF? 或者我自己计算tfidf更简单?

1 个答案:

答案 0 :(得分:0)

您可以使用CountVectorizer代替 HashingTF 。 CountVectorizer也可用于获取术语频率向量。

要使用 CountVectorizer 以及随后的 IDF ,您必须使用DataFrame而不是RDD,因为只有ml包支持CountVectorizer。