Skip to content

Ranker

Base class for ranking models.

Parameters

  • seed (int) – defaults to None

    Random number generation seed. Set this for reproducibility.

Attributes

  • is_contextual

Methods

learn_one

Fits a user-item pair and a real-valued target y.

Parameters

  • user (Union[str, int])
  • item (Union[str, int])
  • y (Union[numbers.Number, bool])
  • x (dict) – defaults to None
predict_one

Predicts the target value of a set of features x.

Parameters

  • user (Union[str, int])
  • item (Union[str, int])
  • x (dict) – defaults to None

Returns

typing.Union[numbers.Number, bool]: The predicted preference from the user for the item.

rank

Rank models by decreasing order of preference for a given user.

Parameters

  • user (Union[str, int])
  • items (Set[Union[str, int]])
  • x (dict) – defaults to None