Package com.r4tings.recommender.model.svd.baseline

Class MeanRatingBaselineModel

java.lang.Object

org.apache.spark.ml.PipelineStage

org.apache.spark.ml.Transformer

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

com.r4tings.recommender.common.ml.AbstractModel

com.r4tings.recommender.common.ml.CommonModel<MeanRatingBaselineModel>

com.r4tings.recommender.model.svd.baseline.MeanRatingBaselineModel

All Implemented Interfaces:

OutputCol, RatingCol, Verbose, java.io.Serializable, org.apache.spark.internal.Logging, org.apache.spark.ml.param.Params, org.apache.spark.ml.util.Identifiable, scala.Serializable

public class MeanRatingBaselineModel
extends CommonModel<MeanRatingBaselineModel>

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from interface org.apache.spark.internal.Logging

org.apache.spark.internal.Logging.SparkShellLoggingFilter

Constructor Summary

Constructors

Constructor	Description
MeanRatingBaselineModel()

Method Summary

Modifier and Type	Method	Description
org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>	getItemBiasDS()
java.lang.Double	getOverallMeanRating()
java.lang.String	getResidualCol()
org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>	getUserBiasDS()
org.apache.spark.ml.param.Param<org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>>	itemBiasDS()
org.apache.spark.ml.param.DoubleParam	overallMeanRating()
org.apache.spark.ml.param.Param<java.lang.String>	residualCol()
protected MeanRatingBaselineModel	self()
MeanRatingBaselineModel	setItemBiasDS(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> value)
MeanRatingBaselineModel	setOverallMeanRating(java.lang.Double value)
MeanRatingBaselineModel	setResidualCol(java.lang.String value)
MeanRatingBaselineModel	setUserBiasDS(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> value)
org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>	transform(org.apache.spark.sql.Dataset<?> ratings)
org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>	transformAll()
org.apache.spark.ml.param.Param<org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>>	userBiasDS()

Methods inherited from class com.r4tings.recommender.common.ml.CommonModel

setItemCol, setOutputCol, setRatingCol, setUserCol, setVerbose

Methods inherited from class com.r4tings.recommender.common.ml.AbstractModel

copy, getItemCol, getOutputCol, getRatingCol, getUserCol, getVerbose, itemCol, outputCol, ratingCol, transformSchema, uid, userCol, verbose

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

hasParent, parent, parent_$eq, setParent

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

transform, transform, transform

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

$, clear, copyValues, copyValues$default$2, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, initializeForcefully, initializeLogIfNecessary, initializeLogIfNecessary, initializeLogIfNecessary$default$2, isDefined, isSet, isTraceEnabled, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning, onParamChange, org$apache$spark$internal$Logging$$log_, org$apache$spark$internal$Logging$$log__$eq, org$apache$spark$ml$param$Params$_setter_$defaultParamMap_$eq, org$apache$spark$ml$param$Params$_setter_$paramMap_$eq, paramMap, params, set, set, set, setDefault, setDefault, toString, transformSchema

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Detail

MeanRatingBaselineModel

public MeanRatingBaselineModel()

Method Detail

transform

public org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> transform(org.apache.spark.sql.Dataset<?> ratings)

Specified by:

transform in class org.apache.spark.ml.Transformer

residualCol

public org.apache.spark.ml.param.Param<java.lang.String> residualCol()

overallMeanRating

public org.apache.spark.ml.param.DoubleParam overallMeanRating()

userBiasDS

public org.apache.spark.ml.param.Param<org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>> userBiasDS()

itemBiasDS

public org.apache.spark.ml.param.Param<org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>> itemBiasDS()

getResidualCol

public java.lang.String getResidualCol()

getOverallMeanRating

public java.lang.Double getOverallMeanRating()

getUserBiasDS

public org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> getUserBiasDS()

getItemBiasDS

public org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> getItemBiasDS()

setResidualCol

public MeanRatingBaselineModel setResidualCol(java.lang.String value)

setOverallMeanRating

public MeanRatingBaselineModel setOverallMeanRating(java.lang.Double value)

setUserBiasDS

public MeanRatingBaselineModel setUserBiasDS(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> value)

setItemBiasDS

public MeanRatingBaselineModel setItemBiasDS(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> value)

self

protected MeanRatingBaselineModel self()

Specified by:

self in class CommonModel<MeanRatingBaselineModel>

transformAll

@Experimental
public org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> transformAll()