choice.LinearFlowUtility

Guide & worked example · Source

choice.LinearFlowUtility(n_features, n_choices, device=None)

Linear-in-features flow utility specification.

The model parameterizes period utility as:

u(x, a) = f(x)^T theta[:, a]

where f(x) is a feature vector for state x.

Attributes

Name	Description
device
theta

Methods

Name	Description
forward	Evaluate flow utility for states and actions.

forward

choice.LinearFlowUtility.forward(states, choice=None, theta=None)

Evaluate flow utility for states and actions.

Args: states: State feature matrix of shape (n_obs, n_features). choice: If provided, returns utility for a single action index. theta: Optional coefficient override, used during optimization.

Returns: Utility tensor of shape (n_obs,) for single-choice mode or (n_obs, n_choices) when choice is None.