Skip to content

Ranker

Base class for ranking models.

Parameters

  • seed

    Type → int | None

    Default → 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 — ID
  • item — ID
  • y — Reward
  • x — dict | None — defaults to None

predict_one

Predicts the target value of a set of features x.

Parameters

  • user — ID
  • item — ID
  • x — dict | None — defaults to None

Returns

Reward: The predicted preference from the user for the item.

rank

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

Parameters

  • user — ID
  • items — set[ID]
  • x — dict | None — defaults to None