Package com.r4tings.recommender.data.normalize

Class ThresholdBinarizer

java.lang.Object

org.apache.spark.ml.PipelineStage

org.apache.spark.ml.Transformer

com.r4tings.recommender.common.ml.AbstractRatingTransformer

com.r4tings.recommender.data.normalize.RatingNormalizer<ThresholdBinarizer>

com.r4tings.recommender.data.normalize.ThresholdBinarizer

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

public class ThresholdBinarizer
extends RatingNormalizer<ThresholdBinarizer>

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
ThresholdBinarizer()

Method Summary

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)
protected org.apache.spark.sql.Dataset<org.apache.spark.sql.Row>	compute(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> ratingDS, java.lang.String groupCol)
java.lang.Double	getThreshold()
protected ThresholdBinarizer	self()
ThresholdBinarizer	setThreshold(java.lang.Double value)
org.apache.spark.ml.param.DoubleParam	threshold()

Methods inherited from class com.r4tings.recommender.data.normalize.RatingNormalizer

getGroup, getGroupCol, getOutputCol, group, scale, setGroup, setGroupCol, setItemCol, setOutputCol, setRatingCol, setUserCol, setVerbose, transform

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

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

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

ThresholdBinarizer

public ThresholdBinarizer()

Method Detail

self

protected ThresholdBinarizer self()

Specified by:

self in class RatingNormalizer<ThresholdBinarizer>

compute

protected org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> compute(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> ratingDS)

Specified by:

compute in class RatingNormalizer<ThresholdBinarizer>

compute

protected org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> compute(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> ratingDS,
                                                                         java.lang.String groupCol)

Specified by:

compute in class RatingNormalizer<ThresholdBinarizer>

threshold

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

getThreshold

public java.lang.Double getThreshold()

setThreshold

public ThresholdBinarizer setThreshold(java.lang.Double value)