Class GeneralMeanRatingBaseline
- java.lang.Object
-
- org.apache.spark.ml.PipelineStage
-
- org.apache.spark.ml.Estimator<AbstractModel>
-
- com.r4tings.recommender.common.ml.AbstractEstimator
-
- com.r4tings.recommender.common.ml.CommonEstimator<GeneralMeanRatingBaseline>
-
- com.r4tings.recommender.model.svd.baseline.GeneralMeanRatingBaseline
-
- 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 GeneralMeanRatingBaseline extends CommonEstimator<GeneralMeanRatingBaseline>
- See Also:
- Serialized Form
-
-
Constructor Summary
Constructors Constructor Description GeneralMeanRatingBaseline()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description MeanRatingBaselineModel
fit(org.apache.spark.sql.Dataset ratings)
java.lang.Integer
getLambda2()
java.lang.Integer
getLambda3()
org.apache.spark.ml.param.IntParam
lambda2()
org.apache.spark.ml.param.IntParam
lambda3()
protected GeneralMeanRatingBaseline
self()
GeneralMeanRatingBaseline
setLambda2(java.lang.Integer value)
GeneralMeanRatingBaseline
setLambda3(java.lang.Integer value)
-
Methods inherited from class com.r4tings.recommender.common.ml.CommonEstimator
setItemCol, setOutputCol, setRatingCol, setUserCol, setVerbose
-
Methods inherited from class com.r4tings.recommender.common.ml.AbstractEstimator
copy, getItemCol, getOutputCol, getRatingCol, getUserCol, getVerbose, itemCol, outputCol, ratingCol, transformSchema, uid, userCol, verbose
-
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
-
-
-
-
Method Detail
-
self
protected GeneralMeanRatingBaseline self()
- Specified by:
self
in classCommonEstimator<GeneralMeanRatingBaseline>
-
fit
public MeanRatingBaselineModel fit(org.apache.spark.sql.Dataset ratings)
- Specified by:
fit
in classorg.apache.spark.ml.Estimator<AbstractModel>
-
lambda2
public org.apache.spark.ml.param.IntParam lambda2()
-
lambda3
public org.apache.spark.ml.param.IntParam lambda3()
-
getLambda2
public java.lang.Integer getLambda2()
-
getLambda3
public java.lang.Integer getLambda3()
-
setLambda2
public GeneralMeanRatingBaseline setLambda2(java.lang.Integer value)
-
setLambda3
public GeneralMeanRatingBaseline setLambda3(java.lang.Integer value)
-
-