Class SingleValueDecompositionModel
- 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<SingleValueDecompositionModel>
-
- com.r4tings.recommender.model.svd.mf.SingleValueDecompositionModel
-
- 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 SingleValueDecompositionModel extends CommonModel<SingleValueDecompositionModel>
- See Also:
- Serialized Form
-
-
Constructor Summary
Constructors Constructor Description SingleValueDecompositionModel()
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description org.apache.spark.ml.param.Param<java.lang.Double[]>featureWeights()java.lang.Double[]getFeatureWeights()org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>getItemFeatureDS()org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>getUserFeatureDS()org.apache.spark.ml.param.Param<org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>>itemFeatureDS()static java.lang.DoublereconstructRating(scala.collection.mutable.WrappedArray<java.lang.Double> u, scala.collection.mutable.WrappedArray<java.lang.Double> s, scala.collection.mutable.WrappedArray<java.lang.Double> v)protected SingleValueDecompositionModelself()SingleValueDecompositionModelsetFeatureWeights(double[] value)SingleValueDecompositionModelsetItemFeatureDS(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> value)SingleValueDecompositionModelsetUserFeatureDS(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.ml.param.Param<org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>>userFeatureDS()-
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.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 SingleValueDecompositionModel self()
- Specified by:
selfin classCommonModel<SingleValueDecompositionModel>
-
transform
public org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> transform(org.apache.spark.sql.Dataset<?> ratings)
- Specified by:
transformin classorg.apache.spark.ml.Transformer
-
reconstructRating
public static java.lang.Double reconstructRating(scala.collection.mutable.WrappedArray<java.lang.Double> u, scala.collection.mutable.WrappedArray<java.lang.Double> s, scala.collection.mutable.WrappedArray<java.lang.Double> v)
-
userFeatureDS
public org.apache.spark.ml.param.Param<org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>> userFeatureDS()
-
featureWeights
public org.apache.spark.ml.param.Param<java.lang.Double[]> featureWeights()
-
itemFeatureDS
public org.apache.spark.ml.param.Param<org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>> itemFeatureDS()
-
getUserFeatureDS
public org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> getUserFeatureDS()
-
getFeatureWeights
public java.lang.Double[] getFeatureWeights()
-
getItemFeatureDS
public org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> getItemFeatureDS()
-
setUserFeatureDS
public SingleValueDecompositionModel setUserFeatureDS(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> value)
-
setItemFeatureDS
public SingleValueDecompositionModel setItemFeatureDS(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> value)
-
setFeatureWeights
public SingleValueDecompositionModel setFeatureWeights(double[] value)
-
-