biogeme.expressions.sparse_log_cross_nested module

Sparse implementation of the cross-nested logit expression.

class biogeme.expressions.sparse_log_cross_nested.SparseLogCrossNested(*args, **kwargs)

Bases: LogCrossNested

CNL expression that skips structurally zero allocation parameters.

recursive_construct_jax_function(numerically_safe)

Generate a JAX function operating only on active memberships.

Return type:

Callable[[Array, Array, Array, Array], Array]

Parameters:

numerically_safe (bool)

uses_sparse_memberships = True