public class ALSModel extends Model<ALSModel>
param: rank rank of the matrix factorization model
param: userFactors a DataFrame that stores user factors in two columns: id
and features
param: itemFactors a DataFrame that stores item factors in two columns: id
and features
Modifier and Type | Method and Description |
---|---|
ALSModel |
copy(ParamMap extra)
Creates a copy of this instance with the same UID and some extra params.
|
java.lang.String |
getItemCol() |
java.lang.String |
getUserCol() |
Param<java.lang.String> |
itemCol()
Param for the column name for item ids.
|
DataFrame |
itemFactors() |
int |
rank() |
ALSModel |
setItemCol(java.lang.String value) |
ALSModel |
setPredictionCol(java.lang.String value) |
ALSModel |
setUserCol(java.lang.String value) |
DataFrame |
transform(DataFrame dataset)
Transforms the input dataset.
|
StructType |
transformSchema(StructType schema)
:: DeveloperApi ::
|
java.lang.String |
uid()
An immutable unique ID for the object and its derivatives.
|
Param<java.lang.String> |
userCol()
Param for the column name for user ids.
|
DataFrame |
userFactors() |
transform, transform, transform
transformSchema
clone, equals, finalize, 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, validateParams
toString
initializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning
public java.lang.String uid()
Identifiable
uid
in interface Identifiable
public int rank()
public DataFrame userFactors()
public DataFrame itemFactors()
public ALSModel setUserCol(java.lang.String value)
public ALSModel setItemCol(java.lang.String value)
public ALSModel setPredictionCol(java.lang.String value)
public DataFrame transform(DataFrame dataset)
Transformer
transform
in class Transformer
dataset
- (undocumented)public StructType transformSchema(StructType schema)
PipelineStage
Derives the output schema from the input schema.
transformSchema
in class PipelineStage
schema
- (undocumented)public ALSModel copy(ParamMap extra)
Params
public Param<java.lang.String> userCol()
public java.lang.String getUserCol()
public Param<java.lang.String> itemCol()
public java.lang.String getItemCol()