当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。