Skip to content

Commit

Permalink
likelihood to new interface. (#114)
Browse files Browse the repository at this point in the history
  • Loading branch information
MaxBlesch authored Jul 16, 2024
1 parent 5a64f3f commit 997cbbf
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/dcegm/likelihood.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,15 +20,13 @@

def create_individual_likelihood_function_for_model(
model: Dict[str, Any],
options: Dict[str, Any],
observed_states: Dict[str, int],
observed_wealth: np.array,
observed_choices: np.array,
exog_savings_grid: np.ndarray,
params_all,
):
solve_func = get_solve_func_for_model(
model=model, exog_savings_grid=exog_savings_grid, options=options
model=model,
)

observed_state_choice_indexes = get_state_choice_index_per_state(
Expand All @@ -37,6 +35,8 @@ def create_individual_likelihood_function_for_model(
state_space_names=model["model_structure"]["state_space_names"],
)

options = model["options"]

# Create the calculation of the choice probabilities, which takes parameters as
# input as well as the solved endogenous wealth grid and the values.
def partial_choice_prob_calculation(value_in, endog_grid_in, params_in):
Expand Down

0 comments on commit 997cbbf

Please sign in to comment.