src.statistical_models package

Submodules

src.statistical_models.mixed_effects module

src.statistical_models.mixed_effects.define_and_return_biogeme(df: DataFrame, alt_spec_vars: dict[int, list[str]], num_classes: int) → BIOGEME

Define and return a Biogeme object for fixed effects models.

Parameters:

• df (pd.DataFrame) – DataFrame containing the data.

• alt_spec_vars (dict[int, list[str]]) – Dictionary mapping alternative IDs to lists of variable names.

src.statistical_models.mixed_effects.define_betas(alt_spec_vars: dict[int, list[str]]) → dict[str, Beta]

Define beta parameters for the fixed effects model.

Parameters:

alt_spec_vars (dict[int, list[str]]) – Dictionary mapping alternative IDs to lists of variable names.

Returns:

Dictionary of beta parameters.

Return type:

dict[str, Beta]

Module contents