dcegm.interpolation.interp1d_dj

Functions

interp1d_policy_and_value_on_wealth_dj(...)

1D interpolation for DJ with consume-all overwrite for policy and value.

interp1d_value_on_wealth_dj(→ jax.numpy.ndarray | float)

1D value interpolation for DJ with consume-all overwrite.

_consume_all_value(→ jax.numpy.ndarray)

Module Contents

dcegm.interpolation.interp1d_dj.interp1d_policy_and_value_on_wealth_dj(wealth: float | jax.numpy.ndarray, wealth_grid: jax.numpy.ndarray, policy_grid: jax.numpy.ndarray, value_grid: jax.numpy.ndarray, compute_utility: Callable, state_choice_vec: Dict[str, int], params: Dict[str, float], discount_factor: float) Tuple[jax.numpy.ndarray | float, jax.numpy.ndarray | float]

1D interpolation for DJ with consume-all overwrite for policy and value.

dcegm.interpolation.interp1d_dj.interp1d_value_on_wealth_dj(wealth: float | jax.numpy.ndarray, wealth_grid: jax.numpy.ndarray, value_grid: jax.numpy.ndarray, compute_utility: Callable, state_choice_vec: Dict[str, int], params: Dict[str, float], discount_factor: float) jax.numpy.ndarray | float

1D value interpolation for DJ with consume-all overwrite.

dcegm.interpolation.interp1d_dj._consume_all_value(wealth: float | jax.numpy.ndarray, value_at_zero_wealth: float | jax.numpy.ndarray, compute_utility: Callable, state_choice_vec: Dict[str, int], params: Dict[str, float], discount_factor: float) jax.numpy.ndarray