calc_diversity_metrics

rectools.metrics.diversity.calc_diversity_metrics(metrics: Dict[str, IntraListDiversity], reco: DataFrame) Dict[str, float][source]

Calculate diversity metrics (only IntraListDiversity now).

Warning: It is not recommended to use this function directly. Use calc_metrics instead.

Parameters
  • metrics (dict(str -> DiversityMetric)) – Dict of metric objects to calculate, where key is metric name and value is metric object.

  • reco (pd.DataFrame) – Recommendations table with columns Columns.User, Columns.Item, Columns.Rank.

Returns

Dictionary where keys are the same with keys in metrics and values are metric calculation results.

Return type

dict(str->float)