biogeme.sampling_of_alternatives.sampling_of_alternatives module

Module in charge of functionalities related to the sampling of alternatives

author:

Michel Bierlaire

date:

Thu Sep 7 10:14:54 2023

class biogeme.sampling_of_alternatives.sampling_of_alternatives.SamplingOfAlternatives(context)[source]

Bases: object

Class dealing with the various methods needed to estimate models with samples of alternatives

Parameters:

context (SamplingContext)

sample_alternatives(chosen)[source]

Performing the sampling of alternatives

Parameters:

chosen (int) – ID of the chosen alternative, that must be included in the choice set.

Return type:

DataFrame

Returns:

data frame containing a sample of alternatives. The first one is the chosen alternative

Raises:

BiogemeError – if the chosen alternative is unknown.

sample_mev_alternatives()[source]

Performing the sampling of alternatives for the MEV terms. Here, the chosen alternative is ignored.

Return type:

DataFrame

Returns:

data frame containing a sample of alternatives

biogeme.sampling_of_alternatives.sampling_of_alternatives.generate_segment_size(sample_size, number_of_segments)[source]

This function calculates the size of each segment, so that they are as close to each other as possible, and cover the full sample

Parameters:
  • sample_size (int) – total size of the sample

  • number_of_segments (int) – number of segments

Returns:

list of length number_of_segments, containing the segment sizes

Return type:

list[int]