本文整理匯總了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]))