當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

Python pyspark FPGrowthModel用法及代碼示例

本文簡要介紹 pyspark.mllib.fpm.FPGrowthModel 的用法。

用法:

class pyspark.mllib.fpm.FPGrowthModel(java_model)

使用並行FP-Growth算法挖掘頻繁項集的FP-Growth模型。

1.4.0 版中的新函數。

例子：

>>> data = [["a", "b", "c"], ["a", "b", "d", "e"], ["a", "c", "e"], ["a", "c", "f"]]
>>> rdd = sc.parallelize(data, 2)
>>> model = FPGrowth.train(rdd, 0.6, 2)
>>> sorted(model.freqItemsets().collect())
[FreqItemset(items=['a'], freq=4), FreqItemset(items=['c'], freq=3), ...
>>> model_path = temp_path + "/fpm"
>>> model.save(sc, model_path)
>>> sameModel = FPGrowthModel.load(sc, model_path)
>>> sorted(model.freqItemsets().collect()) == sorted(sameModel.freqItemsets().collect())
True

相關用法


注：本文由純淨天空篩選整理自spark.apache.org大神的英文原創作品 pyspark.mllib.fpm.FPGrowthModel。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。