RankingEvaluator¶
-
class
RankingEvaluator.
RankingEvaluator
(itemCol=None, k=10, labelCol='label', metricName='ndcgAt', nItems=-1, predictionCol='prediction', ratingCol=None, userCol=None)[source]¶ Bases:
mmlspark.Utils.ComplexParamsMixin
,pyspark.ml.util.JavaMLReadable
,pyspark.ml.util.JavaMLWritable
,pyspark.ml.evaluation.JavaEvaluator
Parameters: - itemCol (str) – Column of items
- k (int) – number of items (default: 10)
- labelCol (str) – label column name (default: label)
- metricName (str) – metric name in evaluation (ndcgAt|map|precisionAtk|recallAtK|diversityAtK|maxDiversity|mrr|fcp) (default: ndcgAt)
- nItems (long) – number of items (default: -1)
- predictionCol (str) – prediction column name (default: prediction)
- ratingCol (str) – Column of ratings
- userCol (str) – Column of users
-
getMetricName
()[source]¶ Returns: metric name in evaluation (ndcgAt|map|precisionAtk|recallAtK|diversityAtK|maxDiversity|mrr|fcp) (default: ndcgAt) Return type: str
-
setMetricName
(value)[source]¶ Parameters: metricName (str) – metric name in evaluation (ndcgAt|map|precisionAtk|recallAtK|diversityAtK|maxDiversity|mrr|fcp) (default: ndcgAt)
-
setParams
(itemCol=None, k=10, labelCol='label', metricName='ndcgAt', nItems=-1, predictionCol='prediction', ratingCol=None, userCol=None)[source]¶ Set the (keyword only) parameters
Parameters: - itemCol (str) – Column of items
- k (int) – number of items (default: 10)
- labelCol (str) – label column name (default: label)
- metricName (str) – metric name in evaluation (ndcgAt|map|precisionAtk|recallAtK|diversityAtK|maxDiversity|mrr|fcp) (default: ndcgAt)
- nItems (long) – number of items (default: -1)
- predictionCol (str) – prediction column name (default: prediction)
- ratingCol (str) – Column of ratings
- userCol (str) – Column of users