recsys_metrics_polars.base#

Classes

BaseMetricAtK(data_info)

BaseRecMetric(data_info)

Base class for metrics

class recsys_metrics_polars.base.BaseRecMetric(data_info)[source]#

Bases: ABC

Base class for metrics

Parameters:

data_info (DataInfo) –

fit(true_interactions, recommendations)[source]#

Prepare data for metric computing

Parameters:
  • true_interactions (DataFrame) – true interactions

  • recommendations (DataFrame) – predicted interactions with scores for each pair query and item

Return type:

BaseRecMetric

abstract compute_per_query(**kwargs)[source]#

Compute metric per query

Return type:

DataFrame

abstract avergae_over_queries(**kwargs)[source]#

Compute mean metric value over all queries

Return type:

float

class recsys_metrics_polars.base.BaseMetricAtK(data_info)[source]#

Bases: BaseRecMetric

Parameters:

data_info (DataInfo) –

abstract compute_per_query(k, **kwargs)[source]#

Compute metric per query

Return type:

DataFrame

Parameters:

k (int) –

abstract avergae_over_queries(k, **kwargs)[source]#

Compute mean metric value over all queries

Return type:

float

Parameters:

k (int) –