public interface FactorizationMachinesParams extends PredictorParams, HasMaxIter, HasStepSize, HasTol, HasSolver, HasSeed, HasFitIntercept, HasRegParam, HasWeightCol
Modifier and Type | Method and Description |
---|---|
IntParam |
factorSize()
Param for dimensionality of the factors (>= 0)
|
BooleanParam |
fitLinear()
Param for whether to fit linear term (aka 1-way term)
|
int |
getFactorSize() |
boolean |
getFitLinear() |
double |
getInitStd() |
double |
getMiniBatchFraction() |
DoubleParam |
initStd()
Param for standard deviation of initial coefficients
|
DoubleParam |
miniBatchFraction()
Param for mini-batch fraction, must be in range (0, 1]
|
Param<String> |
solver()
The solver algorithm for optimization.
|
extractInstances, extractInstances, validateAndTransformSchema
getLabelCol, labelCol
featuresCol, getFeaturesCol
getPredictionCol, predictionCol
clear, copy, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn
toString, uid
getMaxIter, maxIter
getStepSize, stepSize
fitIntercept, getFitIntercept
getRegParam, regParam
getWeightCol, weightCol
IntParam factorSize()
BooleanParam fitLinear()
int getFactorSize()
boolean getFitLinear()
double getInitStd()
double getMiniBatchFraction()
DoubleParam initStd()
DoubleParam miniBatchFraction()