RankingAdapter

class RankingAdapter.RankingAdapter(itemCol=None, k=10, labelCol='label', minRatingsPerItem=1, minRatingsPerUser=1, mode='allUsers', ratingCol=None, recommender=None, userCol=None)[source]

Bases: mmlspark.Utils.ComplexParamsMixin, pyspark.ml.util.JavaMLReadable, pyspark.ml.util.JavaMLWritable, pyspark.ml.wrapper.JavaEstimator

Parameters:
  • itemCol (str) – Column of items
  • k (int) – number of items (default: 10)
  • labelCol (str) – The name of the label column (default: label)
  • minRatingsPerItem (int) – min ratings for items > 0 (default: 1)
  • minRatingsPerUser (int) – min ratings for users > 0 (default: 1)
  • mode (str) – recommendation mode (default: allUsers)
  • ratingCol (str) – Column of ratings
  • recommender (object) – estimator for selection
  • userCol (str) – Column of users
getItemCol()[source]
Returns:Column of items
Return type:str
static getJavaPackage()[source]

Returns package name String.

getK()[source]
Returns:number of items (default: 10)
Return type:int
getLabelCol()[source]
Returns:The name of the label column (default: label)
Return type:str
getMinRatingsPerItem()[source]
Returns:min ratings for items > 0 (default: 1)
Return type:int
getMinRatingsPerUser()[source]
Returns:min ratings for users > 0 (default: 1)
Return type:int
getMode()[source]
Returns:recommendation mode (default: allUsers)
Return type:str
getRatingCol()[source]
Returns:Column of ratings
Return type:str
getRecommender()[source]
Returns:estimator for selection
Return type:object
getUserCol()[source]
Returns:Column of users
Return type:str
classmethod read()[source]

Returns an MLReader instance for this class.

setItemCol(value)[source]
Parameters:itemCol (str) – Column of items
setK(value)[source]
Parameters:k (int) – number of items (default: 10)
setLabelCol(value)[source]
Parameters:labelCol (str) – The name of the label column (default: label)
setMinRatingsPerItem(value)[source]
Parameters:minRatingsPerItem (int) – min ratings for items > 0 (default: 1)
setMinRatingsPerUser(value)[source]
Parameters:minRatingsPerUser (int) – min ratings for users > 0 (default: 1)
setMode(value)[source]
Parameters:mode (str) – recommendation mode (default: allUsers)
setParams(itemCol=None, k=10, labelCol='label', minRatingsPerItem=1, minRatingsPerUser=1, mode='allUsers', ratingCol=None, recommender=None, userCol=None)[source]

Set the (keyword only) parameters

Parameters:
  • itemCol (str) – Column of items
  • k (int) – number of items (default: 10)
  • labelCol (str) – The name of the label column (default: label)
  • minRatingsPerItem (int) – min ratings for items > 0 (default: 1)
  • minRatingsPerUser (int) – min ratings for users > 0 (default: 1)
  • mode (str) – recommendation mode (default: allUsers)
  • ratingCol (str) – Column of ratings
  • recommender (object) – estimator for selection
  • userCol (str) – Column of users
setRatingCol(value)[source]
Parameters:ratingCol (str) – Column of ratings
setRecommender(value)[source]
Parameters:recommender (object) – estimator for selection
setUserCol(value)[source]
Parameters:userCol (str) – Column of users
class RankingAdapter.RankingAdapterModel(java_model=None)[source]

Bases: mmlspark.Utils.ComplexParamsMixin, pyspark.ml.wrapper.JavaModel, pyspark.ml.util.JavaMLWritable, pyspark.ml.util.JavaMLReadable

Model fitted by RankingAdapter.

This class is left empty on purpose. All necessary methods are exposed through inheritance.

static getJavaPackage()[source]

Returns package name String.

classmethod read()[source]

Returns an MLReader instance for this class.