biogeme.calculator.multiple_formula module

This module defines the MultiRowEvaluator class, which evaluates multiple expressions on a given database using JAX for efficient batched computation. It returns results as a pandas DataFrame with one column per expression and one row per observation.

Michel Bierlaire Wed Apr 2 13:10:17 2025

class biogeme.calculator.multiple_formula.MultiRowEvaluator(model_elements, numerically_safe, use_jit)

Bases: object

Evaluates multiple expressions on a common dataset using JAX and returns results as a pandas DataFrame. This class compiles all expressions into JAX functions and evaluates them efficiently in a single batched operation.

Parameters:

• model_elements (ModelElements) – Object containing the expressions and all elements needed to calculate them.

• numerically_safe (bool)

• use_jit (bool)

evaluate(the_betas)

Evaluates all expressions using the provided beta values.

Parameters:

the_betas (dict[str, float]) – A dictionary mapping beta names to their numerical values.

Return type:

DataFrame

Returns:

A pandas DataFrame with one column per expression and one row per observation.