public interface ALSParams extends ALSModelParams, HasMaxIter, HasRegParam, HasCheckpointInterval, HasSeed
Modifier and Type | Method and Description |
---|---|
DoubleParam |
alpha()
Param for the alpha parameter in the implicit preference formulation (nonnegative).
|
Param<String> |
finalStorageLevel()
Param for StorageLevel for ALS model factors.
|
double |
getAlpha() |
String |
getFinalStorageLevel() |
boolean |
getImplicitPrefs() |
String |
getIntermediateStorageLevel() |
boolean |
getNonnegative() |
int |
getNumItemBlocks() |
int |
getNumUserBlocks() |
int |
getRank() |
String |
getRatingCol() |
BooleanParam |
implicitPrefs()
Param to decide whether to use implicit preference.
|
Param<String> |
intermediateStorageLevel()
Param for StorageLevel for intermediate datasets.
|
BooleanParam |
nonnegative()
Param for whether to apply nonnegativity constraints.
|
IntParam |
numItemBlocks()
Param for number of item blocks (positive).
|
IntParam |
numUserBlocks()
Param for number of user blocks (positive).
|
IntParam |
rank()
Param for rank of the matrix factorization (positive).
|
Param<String> |
ratingCol()
Param for the column name for ratings.
|
StructType |
validateAndTransformSchema(StructType schema)
Validates and transforms the input schema.
|
checkedCast, coldStartStrategy, getColdStartStrategy, getItemCol, getUserCol, itemCol, userCol
getPredictionCol, predictionCol
blockSize, getBlockSize
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
getRegParam, regParam
checkpointInterval, getCheckpointInterval
DoubleParam alpha()
Param<String> finalStorageLevel()
StorageLevel
.
Default: "MEMORY_AND_DISK".
double getAlpha()
String getFinalStorageLevel()
boolean getImplicitPrefs()
String getIntermediateStorageLevel()
boolean getNonnegative()
int getNumItemBlocks()
int getNumUserBlocks()
int getRank()
String getRatingCol()
BooleanParam implicitPrefs()
Param<String> intermediateStorageLevel()
StorageLevel
. Cannot be "NONE".
Default: "MEMORY_AND_DISK".
BooleanParam nonnegative()
IntParam numItemBlocks()
IntParam numUserBlocks()
IntParam rank()
Param<String> ratingCol()
StructType validateAndTransformSchema(StructType schema)
schema
- input schema