public class FPGrowth extends Estimator<FPGrowthModel> implements DefaultParamsWritable
Modifier and Type | Method and Description |
---|---|
static Params |
clear(Param<?> param) |
FPGrowth |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params.
|
static String |
explainParam(Param<?> param) |
static String |
explainParams() |
static ParamMap |
extractParamMap() |
static ParamMap |
extractParamMap(ParamMap extra) |
FPGrowthModel |
fit(Dataset<?> dataset)
Fits a model to the input data.
|
static <T> scala.Option<T> |
get(Param<T> param) |
static <T> scala.Option<T> |
getDefault(Param<T> param) |
static String |
getItemsCol() |
String |
getItemsCol() |
static double |
getMinConfidence() |
double |
getMinConfidence() |
static double |
getMinSupport() |
double |
getMinSupport() |
static int |
getNumPartitions() |
int |
getNumPartitions() |
static <T> T |
getOrDefault(Param<T> param) |
static Param<Object> |
getParam(String paramName) |
static String |
getPredictionCol() |
static <T> boolean |
hasDefault(Param<T> param) |
static boolean |
hasParam(String paramName) |
static boolean |
isDefined(Param<?> param) |
static boolean |
isSet(Param<?> param) |
static Param<String> |
itemsCol() |
Param<String> |
itemsCol()
Items column name.
|
static FPGrowth |
load(String path) |
static DoubleParam |
minConfidence() |
DoubleParam |
minConfidence()
Minimal confidence for generating Association Rule.
|
static DoubleParam |
minSupport() |
DoubleParam |
minSupport()
Minimal support level of the frequent pattern.
|
static IntParam |
numPartitions() |
IntParam |
numPartitions()
Number of partitions (at least 1) used by parallel FP-growth.
|
static Param<?>[] |
params() |
static Param<String> |
predictionCol() |
static void |
save(String path) |
static <T> Params |
set(Param<T> param,
T value) |
FPGrowth |
setItemsCol(String value) |
FPGrowth |
setMinConfidence(double value) |
FPGrowth |
setMinSupport(double value) |
FPGrowth |
setNumPartitions(int value) |
FPGrowth |
setPredictionCol(String value) |
static String |
toString() |
StructType |
transformSchema(StructType schema)
:: DeveloperApi ::
|
String |
uid()
An immutable unique ID for the object and its derivatives.
|
StructType |
validateAndTransformSchema(StructType schema)
Validates and transforms the input schema.
|
static MLWriter |
write() |
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
clear, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn
toString
write
save
initializeLogging, initializeLogIfNecessary, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning
public static FPGrowth load(String path)
public static String toString()
public static Param<?>[] params()
public static String explainParam(Param<?> param)
public static String explainParams()
public static final boolean isSet(Param<?> param)
public static final boolean isDefined(Param<?> param)
public static boolean hasParam(String paramName)
public static Param<Object> getParam(String paramName)
public static final <T> scala.Option<T> get(Param<T> param)
public static final <T> T getOrDefault(Param<T> param)
public static final <T> scala.Option<T> getDefault(Param<T> param)
public static final <T> boolean hasDefault(Param<T> param)
public static final ParamMap extractParamMap()
public static final Param<String> predictionCol()
public static final String getPredictionCol()
public static Param<String> itemsCol()
public static String getItemsCol()
public static DoubleParam minSupport()
public static double getMinSupport()
public static IntParam numPartitions()
public static int getNumPartitions()
public static DoubleParam minConfidence()
public static double getMinConfidence()
public static void save(String path) throws java.io.IOException
java.io.IOException
public static MLWriter write()
public String uid()
Identifiable
uid
in interface Identifiable
public FPGrowth setMinSupport(double value)
public FPGrowth setNumPartitions(int value)
public FPGrowth setMinConfidence(double value)
public FPGrowth setItemsCol(String value)
public FPGrowth setPredictionCol(String value)
public FPGrowthModel fit(Dataset<?> dataset)
Estimator
fit
in class Estimator<FPGrowthModel>
dataset
- (undocumented)public StructType transformSchema(StructType schema)
PipelineStage
Check transform validity and derive the output schema from the input schema.
We check validity for interactions between parameters during transformSchema
and
raise an exception if any parameter value is invalid. Parameter value checks which
do not depend on other parameters are handled by Param.validate()
.
Typical implementation should first conduct verification on schema change and parameter validity, including complex parameter interaction checks.
transformSchema
in class PipelineStage
schema
- (undocumented)public FPGrowth copy(ParamMap extra)
Params
defaultCopy()
.copy
in interface Params
copy
in class Estimator<FPGrowthModel>
extra
- (undocumented)public Param<String> itemsCol()
public String getItemsCol()
public DoubleParam minSupport()
public double getMinSupport()
public IntParam numPartitions()
public int getNumPartitions()
public DoubleParam minConfidence()
public double getMinConfidence()
public StructType validateAndTransformSchema(StructType schema)
schema
- input schema