Class RatingSimilarityMeasurer<T extends RatingSimilarityMeasurer<T>>
- java.lang.Object
-
- org.apache.spark.ml.PipelineStage
-
- org.apache.spark.ml.Transformer
-
- com.r4tings.recommender.common.ml.AbstractRatingTransformer
-
- com.r4tings.recommender.model.measures.similarity.RatingSimilarityMeasurer<T>
-
- All Implemented Interfaces:
GroupCol,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
- Direct Known Subclasses:
CosineSimilarityMeasurer,EuclideanSimilarityMeasurer,ExtendedJaccardSimilarityMeasurer,JaccardSimilarityMeasurer,ManhattanSimilarityMeasurer,PearsonSimilarityMeasurer,SimpleMatchingSimilarityMeasurer
public abstract class RatingSimilarityMeasurer<T extends RatingSimilarityMeasurer<T>> extends AbstractRatingTransformer implements GroupCol
- See Also:
- Serialized Form
-
-
Constructor Summary
Constructors Constructor Description RatingSimilarityMeasurer(java.lang.String prefix)
-
Method Summary
All Methods Instance Methods Abstract Methods Concrete Methods Modifier and Type Method Description protected org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>compute(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> ratingDS, org.apache.spark.sql.Column outputColumn)protected abstract org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>execute(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> ratingDS)GroupgetGroup()java.lang.StringgetGroupCol()java.lang.Object[]getIds()java.lang.BooleangetImputeZero()org.apache.spark.ml.param.Param<Group>group()org.apache.spark.ml.param.Param<java.lang.Object[]>ids()org.apache.spark.ml.param.BooleanParamimputeZero()protected abstract Tself()TsetGroup(Group value)TsetGroupCol(java.lang.String value)TsetIds(java.lang.Object... value)TsetImputeZero(java.lang.Boolean value)TsetItemCol(java.lang.String value)TsetOutputCol(java.lang.String value)TsetRatingCol(java.lang.String value)TsetUserCol(java.lang.String value)TsetVerbose(java.lang.Boolean value)org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>transform(org.apache.spark.sql.Dataset<?> ratings)-
Methods inherited from class com.r4tings.recommender.common.ml.AbstractRatingTransformer
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
-
transform
public final 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
-
group
public org.apache.spark.ml.param.Param<Group> group()
-
imputeZero
public org.apache.spark.ml.param.BooleanParam imputeZero()
-
ids
public org.apache.spark.ml.param.Param<java.lang.Object[]> ids()
-
getGroupCol
public java.lang.String getGroupCol()
- Specified by:
getGroupColin interfaceGroupCol
-
getImputeZero
public java.lang.Boolean getImputeZero()
-
getIds
public java.lang.Object[] getIds()
-
setVerbose
public T setVerbose(java.lang.Boolean value)
- Specified by:
setVerbosein interfaceVerbose
-
setOutputCol
public T setOutputCol(java.lang.String value)
- Specified by:
setOutputColin interfaceOutputCol
-
setUserCol
public T setUserCol(java.lang.String value)
- Specified by:
setUserColin interfaceRatingCol
-
setItemCol
public T setItemCol(java.lang.String value)
- Specified by:
setItemColin interfaceRatingCol
-
setRatingCol
public T setRatingCol(java.lang.String value)
- Specified by:
setRatingColin interfaceRatingCol
-
setGroupCol
public T setGroupCol(java.lang.String value)
- Specified by:
setGroupColin interfaceGroupCol
-
setImputeZero
public T setImputeZero(java.lang.Boolean value)
-
setIds
public T setIds(java.lang.Object... value)
-
compute
protected org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> compute(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> ratingDS, org.apache.spark.sql.Column outputColumn)
-
self
protected abstract T self()
-
execute
protected abstract org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> execute(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> ratingDS)
-
-