org.apache.spark.ml.recommendation

Class ALSModel

Object

org.apache.spark.ml.PipelineStage

org.apache.spark.ml.Transformer

org.apache.spark.ml.Model<ALSModel>

org.apache.spark.ml.recommendation.ALSModel

All Implemented Interfaces:

java.io.Serializable, Logging, Identifiable, HasMaxIter, HasPredictionCol, HasRegParam, Params, ALSParams

public class ALSModel
extends Model<ALSModel>
implements ALSParams

Model fitted by ALS.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
ALSModel(ALS parent, ParamMap fittingParamMap, int k, RDD<scala.Tuple2<Object,float[]>> userFactors, RDD<scala.Tuple2<Object,float[]>> itemFactors)

Method Summary

Methods

Modifier and Type	Method and Description
ParamMap	fittingParamMap() Fitting parameters, such that parent.fit(..., fittingParamMap) could reproduce the model.
ALS	parent() The parent estimator that produced this model.
ALSModel	setPredictionCol(String value)
DataFrame	transform(DataFrame dataset, ParamMap paramMap) Transforms the dataset with provided parameter map as additional parameters.
org.apache.spark.sql.types.StructType	transformSchema(org.apache.spark.sql.types.StructType schema, ParamMap paramMap) :: DeveloperAPI ::

Methods inherited from class org.apache.spark.ml.Transformer

transform, transform

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.apache.spark.ml.recommendation.ALSParams

alpha, getAlpha, getImplicitPrefs, getItemCol, getNonnegative, getNumItemBlocks, getNumUserBlocks, getRank, getRatingCol, getUserCol, implicitPrefs, itemCol, nonnegative, numItemBlocks, numUserBlocks, rank, ratingCol, userCol, validateAndTransformSchema

Methods inherited from interface org.apache.spark.ml.param.HasMaxIter

getMaxIter, maxIter

Methods inherited from interface org.apache.spark.ml.param.HasRegParam

getRegParam, regParam

Methods inherited from interface org.apache.spark.ml.param.HasPredictionCol

getPredictionCol, predictionCol

Methods inherited from interface org.apache.spark.ml.param.Params

addOutputColumn, checkInputColumn, explainParams, get, getParam, isSet, paramMap, params, set, set, validate, validate

Methods inherited from interface org.apache.spark.ml.Identifiable

uid

Methods inherited from interface org.apache.spark.Logging

initializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning

Constructor Detail

ALSModel

public ALSModel(ALS parent,
        ParamMap fittingParamMap,
        int k,
        RDD<scala.Tuple2<Object,float[]>> userFactors,
        RDD<scala.Tuple2<Object,float[]>> itemFactors)

Method Detail

parent

public ALS parent()

Description copied from class: Model

The parent estimator that produced this model.

Specified by:

parent in class Model<ALSModel>

fittingParamMap

public ParamMap fittingParamMap()

Description copied from class: Model

Fitting parameters, such that parent.fit(..., fittingParamMap) could reproduce the model.

Specified by:

fittingParamMap in class Model<ALSModel>

setPredictionCol

public ALSModel setPredictionCol(String value)

transform

public DataFrame transform(DataFrame dataset,
                  ParamMap paramMap)

Description copied from class: Transformer

Transforms the dataset with provided parameter map as additional parameters.

Specified by:

transform in class Transformer

Parameters:

dataset - input dataset

paramMap - additional parameters, overwrite embedded params

Returns:

transformed dataset

transformSchema

public org.apache.spark.sql.types.StructType transformSchema(org.apache.spark.sql.types.StructType schema,
                                                    ParamMap paramMap)

Description copied from class: PipelineStage

:: DeveloperAPI ::

Derives the output schema from the input schema and parameters. The schema describes the columns and types of the data.

Specified by:

transformSchema in class PipelineStage

Parameters:

schema - Input schema to this stage

paramMap - Parameters passed to this stage

Returns:

Output schema from this stage