Index

_

__init__() (src.models.base_model.ElectionForecastModel method) (src.models.hierarchical_bayes.HierarchicalBayesModel method) (src.models.improved_kalman.ImprovedKalmanModel method) (src.models.kalman_diffusion.KalmanDiffusionModel method) (src.models.poll_average.PollAverageModel method)

C

compute_metrics() (in module src.utils.data_utils)

D

discover_models() (in module src.scripts.generate_plots) (in module src.scripts.run_all_models)

E

ElectionForecastModel (class in src.models.base_model)

estimate_house_effects() (src.models.hierarchical_bayes.HierarchicalBayesModel method)

F

fit_and_forecast() (src.models.base_model.ElectionForecastModel method) (src.models.hierarchical_bayes.HierarchicalBayesModel method) (src.models.improved_kalman.ImprovedKalmanModel method) (src.models.kalman_diffusion.KalmanDiffusionModel method) (src.models.poll_average.PollAverageModel method)

fit_state_diffusion() (src.models.kalman_diffusion.KalmanDiffusionModel method)

G

generate_forecast_dates() (in module src.scripts.run_all_models) get_current_election_date() (in module src.utils.data_utils)

get_election_date() (in module src.utils.data_utils) get_state_list() (in module src.utils.data_utils)

H

HierarchicalBayesModel (class in src.models.hierarchical_bayes)

I

ImprovedKalmanModel (class in src.models.improved_kalman)

K

kalman_filter_rts() (src.models.hierarchical_bayes.HierarchicalBayesModel method) (src.models.improved_kalman.ImprovedKalmanModel method)

kalman_filter_smoother() (src.models.kalman_diffusion.KalmanDiffusionModel method) KalmanDiffusionModel (class in src.models.kalman_diffusion)

L

load_data() (src.models.base_model.ElectionForecastModel method) load_election_results() (in module src.utils.data_utils)

load_fundamentals() (in module src.utils.data_utils) load_polling_data() (in module src.utils.data_utils)

M

main() (in module src.scripts.compare_models) (in module src.scripts.generate_plots) (in module src.scripts.run_all) (in module src.scripts.run_all_models) module src.models.base_model src.models.hierarchical_bayes src.models.improved_kalman src.models.kalman_diffusion src.models.poll_average src.scripts.compare_models src.scripts.generate_plots src.scripts.run_all src.scripts.run_all_models src.utils.data_utils

P

parse_metrics() (in module src.scripts.compare_models)

plot_state() (src.models.base_model.ElectionForecastModel method) PollAverageModel (class in src.models.poll_average)

R

run_forecast() (src.models.base_model.ElectionForecastModel method)

run_step() (in module src.scripts.run_all) run_with_temp_argv() (in module src.scripts.run_all)

S

save_results() (src.models.base_model.ElectionForecastModel method) set_election_config() (in module src.utils.data_utils) simulate_forward() (src.models.improved_kalman.ImprovedKalmanModel method) (src.models.kalman_diffusion.KalmanDiffusionModel method) src.models.base_model module src.models.hierarchical_bayes module src.models.improved_kalman module src.models.kalman_diffusion module

src.models.poll_average module src.scripts.compare_models module src.scripts.generate_plots module src.scripts.run_all module src.scripts.run_all_models module src.utils.data_utils module