org.apache.spark.mllib.fpm

Class FPGrowthModel<Item>

Object

org.apache.spark.mllib.fpm.FPGrowthModel<Item>

All Implemented Interfaces:

java.io.Serializable, Saveable

public class FPGrowthModel<Item>
extends Object
implements Saveable, scala.Serializable

Model trained by FPGrowth, which holds frequent itemsets. param: freqItemsets frequent itemset, which is an RDD of FreqItemset

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	FPGrowthModel.SaveLoadV1_0$

Constructor Summary

Constructors

Constructor and Description
FPGrowthModel(RDD<FPGrowth.FreqItemset<Item>> freqItemsets, scala.reflect.ClassTag<Item> evidence$1)

Method Summary

Methods

Modifier and Type	Method and Description
RDD<FPGrowth.FreqItemset<Item>>	freqItemsets()
RDD<AssociationRules.Rule<Item>>	generateAssociationRules(double confidence) Generates association rules for the Items in freqItemsets.
static FPGrowthModel<?>	load(SparkContext sc, String path)
void	save(SparkContext sc, String path) Save this model to the given path.

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

FPGrowthModel

public FPGrowthModel(RDD<FPGrowth.FreqItemset<Item>> freqItemsets,
             scala.reflect.ClassTag<Item> evidence$1)

Method Detail

load

public static FPGrowthModel<?> load(SparkContext sc,
                    String path)

freqItemsets

public RDD<FPGrowth.FreqItemset<Item>> freqItemsets()

generateAssociationRules

public RDD<AssociationRules.Rule<Item>> generateAssociationRules(double confidence)

Generates association rules for the Items in freqItemsets.

Parameters:

confidence - minimal confidence of the rules produced

Returns:

(undocumented)

save

public void save(SparkContext sc,
        String path)

Save this model to the given path. It only works for Item datatypes supported by DataFrames.

This saves: - human-readable (JSON) model metadata to path/metadata/ - Parquet formatted data to path/data/

The model may be loaded using FPGrowthModel.load.

Specified by:

save in interface Saveable

Parameters:

sc - Spark context used to save model data.

path - Path specifying the directory in which to save this model. If the directory already exists, this method throws an exception.