VowpalSoftmaxLearner

class coba.learners.VowpalSoftmaxLearner

Softmax exploration with a VW contextual bandit learner.

More information on VW exploration algorithms can be found here.

Constructors

__init__(softmax: float = 10, features: Sequence[str] = (1, 'a', 'ax', 'axx'), seed: int | None = 1, **kwargs) → None

Instantiate a VowpalSoftmaxLearner.

Parameters:

  • softmax – An exploration parameter with 0 indicating predictions should be completely random and infinity indicating that predictions should be greedy. For more information see lambda.

  • features – A list of namespaces and interactions to use when learning reward functions.

  • seed – The seed used by VW to generate any necessary randomness.

  • kwargs – Additional key-word args are passed on as VW CLI arguments (unless removed in the function).

Methods

finish()

Finish all pending work (e.g., write buffers to disk).

learn(context: Context, action: Action, reward: float, probability: float, actions: Actions | None = None) → None

Learn about the action taken in the context.

Parameters:

  • context – The context in which the action was taken.

  • action – The action that was taken.

  • reward – The reward for the given context and action (feedback for IGL problems).

  • probability – The probability the given action was taken.

  • **kwargs – Optional information returned during prediction.

predict(context: Context, actions: Actions) → Tuple[Action, Prob, Kwargs]

Predict which action to take in the context.

Parameters:

  • context – The current context. It will either be None (multi-armed bandit), a value (a single feature), a sequence of values (dense features), or a dictionary (sparse features).

  • actions – The current set of actions to choose from in the given context. Each action will either be a value (a single feature), a sequence of values (dense features), or a dictionary (sparse features).

Returns:

A Prediction. Several prediction formats are supported. See the type-hint for these.

score(context: Context, actions: Actions, action: Action) → Prob

Propensity score an action.

Parameters:

  • context – The current context.

  • actions – The current set of actions that can be chosen.

  • action – The action to propensity score.

Returns:

The propensity score of the given action. That is, P(action|context,actions).

Attributes

params