traveler.demo.tour_mode.simple_choice.probability

simple_choice.probability(store, *, choosers=None, param=None)

Calculate the choice probabilities using a simple choice model.

Parameters:

• choosers (JaxTable) – A table containing choosers data.

• store (AttrPyTree) – A store containing additional data such as skims.

• param (dict[str, float]) – A dictionary of parameters for the utility function.

Returns:

An array of choice probabilities for each alternative.

Return type:

jax.Array