本文整理汇总了Python中pyspark.ml.feature.HashingTF.setInputCol方法的典型用法代码示例。如果您正苦于以下问题:Python HashingTF.setInputCol方法的具体用法?Python HashingTF.setInputCol怎么用?Python HashingTF.setInputCol使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类pyspark.ml.feature.HashingTF
的用法示例。
在下文中一共展示了HashingTF.setInputCol方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_apply_binary_term_freqs
# 需要导入模块: from pyspark.ml.feature import HashingTF [as 别名]
# 或者: from pyspark.ml.feature.HashingTF import setInputCol [as 别名]
def test_apply_binary_term_freqs(self):
df = self.spark.createDataFrame([(0, ["a", "a", "b", "c", "c", "c"])], ["id", "words"])
n = 10
hashingTF = HashingTF()
hashingTF.setInputCol("words").setOutputCol("features").setNumFeatures(n).setBinary(True)
output = hashingTF.transform(df)
features = output.select("features").first().features.toArray()
expected = Vectors.dense([1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]).toArray()
for i in range(0, n):
self.assertAlmostEqual(features[i], expected[i], 14, "Error at " + str(i) +
": expected " + str(expected[i]) + ", got " + str(features[i]))