public class ALSModel extends Model<ALSModel> implements MLWritable
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 |
|---|---|
UserDefinedFunction |
checkedCast()
Attempts to safely cast a user/item id to an Int.
|
static Params |
clear(Param<?> param) |
ALSModel |
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) |
static <T> scala.Option<T> |
get(Param<T> param) |
static <T> scala.Option<T> |
getDefault(Param<T> param) |
static String |
getItemCol() |
String |
getItemCol() |
static <T> T |
getOrDefault(Param<T> param) |
static Param<Object> |
getParam(String paramName) |
static String |
getPredictionCol() |
static String |
getUserCol() |
String |
getUserCol() |
static <T> boolean |
hasDefault(Param<T> param) |
static boolean |
hasParam(String paramName) |
static boolean |
hasParent() |
static boolean |
isDefined(Param<?> param) |
static boolean |
isSet(Param<?> param) |
static Param<String> |
itemCol() |
Param<String> |
itemCol()
Param for the column name for item ids.
|
Dataset<Row> |
itemFactors() |
static ALSModel |
load(String path) |
static Param<?>[] |
params() |
static void |
parent_$eq(Estimator<M> x$1) |
static Estimator<M> |
parent() |
static Param<String> |
predictionCol() |
int |
rank() |
static MLReader<ALSModel> |
read() |
static void |
save(String path) |
static <T> Params |
set(Param<T> param,
T value) |
ALSModel |
setItemCol(String value) |
static M |
setParent(Estimator<M> parent) |
ALSModel |
setPredictionCol(String value) |
ALSModel |
setUserCol(String value) |
static String |
toString() |
Dataset<Row> |
transform(Dataset<?> dataset)
Transforms the input dataset.
|
StructType |
transformSchema(StructType schema)
:: DeveloperApi ::
|
String |
uid()
An immutable unique ID for the object and its derivatives.
|
static Param<String> |
userCol() |
Param<String> |
userCol()
Param for the column name for user ids.
|
Dataset<Row> |
userFactors() |
MLWriter |
write()
Returns an
MLWriter instance for this ML instance. |
transform, transform, transformequals, getClass, hashCode, notify, notifyAll, toString, wait, wait, waitclear, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, shouldOwntoStringsavepublic static ALSModel 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 Estimator<M> parent()
public static void parent_$eq(Estimator<M> x$1)
public static M setParent(Estimator<M> parent)
public static boolean hasParent()
public static final Param<String> predictionCol()
public static final String getPredictionCol()
public static Param<String> userCol()
public static String getUserCol()
public static Param<String> itemCol()
public static String getItemCol()
public static void save(String path)
throws java.io.IOException
java.io.IOExceptionpublic String uid()
Identifiableuid in interface Identifiablepublic int rank()
public ALSModel setUserCol(String value)
public ALSModel setItemCol(String value)
public ALSModel setPredictionCol(String value)
public Dataset<Row> transform(Dataset<?> dataset)
Transformertransform in class Transformerdataset - (undocumented)public StructType transformSchema(StructType schema)
PipelineStageCheck 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 PipelineStageschema - (undocumented)public ALSModel copy(ParamMap extra)
ParamsdefaultCopy().public MLWriter write()
MLWritableMLWriter instance for this ML instance.write in interface MLWritablepublic Param<String> userCol()
public String getUserCol()
public Param<String> itemCol()
public String getItemCol()
public UserDefinedFunction checkedCast()