def get_model_spec_time_varying_with_covariates(covariate_delay): """Return model specification with hyperparameters and covariates.""" result = model_spec_lib.ModelSpec( hparams={ "initial_learning_rate": 0.003061242915556316, "momentum": 0.2, "decay_steps": 1000, "fine_tuning_steps": 100, "fine_tuning_decay": 1.0, "decay_rate": 1.0, "location_dependent_init": False, "infected_threshold": 3, "restart_threshold": 1000, "time_scale_weight": 0.00006243159539906051, "train_loss_coefs": [0, 0.001, 0.2, 0.1, 0.01, 0.01, 0.2, 0.1, 0.01], "valid_loss_coefs": [0, 0.001, 0.2, 0.1, 0.01, 0.01, 0.2, 0.1, 0.01], "sync_coef": 1.0, "reduced_sync_undoc": 1.0, "smooth_coef": 0.5, "first_dose_vaccine_ratio_per_day_init": 0.0, "second_dose_vaccine_ratio_per_day_init": 0.0, "average_contact_id_rate_init": -1.9131825459930378, "average_contact_iud_rate_init": -1.071866945725303, "reinfectable_rate_init": -5.548940468292865, "alpha_rate_init": -2.272765554778715, "diagnosis_rate_init": -2.095597433974376, "recovery_id_rate_init": -1.495660223962899, "recovery_iud_rate_init": -1.475605314236803, "recovery_h_rate_init": -1.9032896753850963, "hospitalization_rate_init": -1.4331763640928012, "death_id_rate_init": -1.8060447968974489, "death_h_rate_init": -1.4886876719378206, "bias_penalty_coef": 0.2835406167308398, "r_eff_penalty_coef": 2.0, "acceleration_death_coef": 0.1, "acceleration_confirm_coef": 0.1, "acceleration_hospital_coef": 0.1, "quantile_encoding_window": 7, "quantile_smooth_coef": 0.9, "quantile_training_iteration_ratio": 0.5, "width_coef_train": 10.0, "width_coef_valid": 5.0, "quantile_cum_viol_coef": 500.0, "increment_loss_weight": 0.0, "lasso_penalty_coef": 1.0, "covariate_training_mixing_coef": 1.0, "train_window_range": 2.0, "partial_mean_interval": 4, "direction_loss_coef": 2000.0, "train_crps_weight": 0.25, "valid_crps_weight": 0.25 }, encoder_specs=[ model_spec_lib.EncoderSpec( encoder_name="first_dose_vaccine_ratio_per_day", encoder_type="vaccine", vaccine_type="first_dose", covariate_feature_specs=[ first_dose_vaccine_ratio_per_day, first_dose_vaccine_effectiveness, second_dose_vaccine_effectiveness ], ), model_spec_lib.EncoderSpec( encoder_name="second_dose_vaccine_ratio_per_day", encoder_type="vaccine", vaccine_type="second_dose", covariate_feature_specs=[ second_dose_vaccine_ratio_per_day, first_dose_vaccine_effectiveness, second_dose_vaccine_effectiveness ], ), model_spec_lib.EncoderSpec( encoder_name="average_contact_id_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, income_per_capita, aqi_mean_2018, household, ] + extra_static_features, covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, mobility_index, mobility_samples, confirmed_per_csrp_tests ] + extra_covariates + npi_features, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="average_contact_iud_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, income_per_capita, aqi_mean_2018, household, ] + extra_static_features, covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, mobility_index, mobility_samples, confirmed_per_csrp_tests ] + extra_covariates + npi_features, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="reinfectable_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, household_food_stamp, income_per_capita, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, rating_average, aqi_mean_2018, kaiser_60plus_population ] + extra_static_features, covariate_feature_specs=None, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, }), model_spec_lib.EncoderSpec(encoder_name="alpha_rate", encoder_type="gam", static_feature_specs=[], covariate_feature_specs=None, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, }), model_spec_lib.EncoderSpec( encoder_name="diagnosis_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, household_food_stamp, income_per_capita, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, rating_average, aqi_mean_2018, kaiser_60plus_population ] + extra_static_features, covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, confirmed_per_csrp_tests, total_tests_per_capita, dow_feature, ] + extra_covariates, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="recovery_id_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, household_food_stamp, income_per_capita, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, rating_average, aqi_mean_2018, kaiser_60plus_population ] + extra_static_features, covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, ] + extra_covariates, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="recovery_iud_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, household_food_stamp, income_per_capita, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, rating_average, aqi_mean_2018, kaiser_60plus_population ] + extra_static_features, covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum ] + extra_covariates, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="recovery_h_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, household_food_stamp, income_per_capita, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, rating_average, aqi_mean_2018, kaiser_60plus_population, icu_beds ] + extra_static_features, covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, ] + extra_covariates, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="hospitalization_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, household_food_stamp, income_per_capita, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, rating_average, aqi_mean_2018, kaiser_60plus_population ] + extra_static_features, covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, confirmed_per_csrp_tests, total_tests_per_capita, ] + extra_covariates, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="death_id_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, household_food_stamp, income_per_capita, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, rating_average, aqi_mean_2018, kaiser_60plus_population, icu_beds ] + extra_static_features, covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, ] + extra_covariates, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="death_h_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, household_food_stamp, income_per_capita, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, rating_average, aqi_mean_2018, kaiser_60plus_population, icu_beds ] + extra_static_features, covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, ] + extra_covariates, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), ]) return result
STATIC_MODEL = model_spec_lib.ModelSpec( hparams={ "initial_learning_rate": 0.001338395965177607, "decay_steps": 5, "decay_rate": 0.99, "fine_tuning_steps": 1000, "location_dependent_init": True, "infected_threshold": 10, "restart_threshold": 300, "time_scale_weight": 0.000021644778644786284, "train_loss_coefs": [0, 0, 1, 0.0011310082312404356], "valid_loss_coefs": [0, 0, 1, 0], "sync_coef": 0.5, "first_dose_vaccine_ratio_per_day_init": 0.0, "second_dose_vaccine_ratio_per_day_init": 0.0, "first_dose_vaccine_effectiveness_init": 0.0, "second_dose_vaccine_effectiveness_init": 0.0, "average_contact_id_rate_init": -2.4387148524945594, "average_contact_iud_rate_init": -1.561117964142531, "reinfectable_rate_init": -5.447821316098564, "alpha_rate_init": -1.1906569150841058, "diagnosis_rate_init": -1.589940307765024, "recovery_id_rate_init": -1.5723201455595701, "recovery_iud_rate_init": -1.8295556397198884, "recovery_h_rate_init": -1.8295556397198884, "death_id_rate_init": -1.0766994750650696, "death_h_rate_init": -1.0766994750650696, "compartmental_penalty_coef": 96.12757205042173, "variable_smoothness_penalty_coef": 5.988152939976956, "bias_penalty_coef": 0.27315803186400106, "boundary_penalty_coef": 2.600265396382176, "lasso_penalty_coef": 1.0, "r_eff_penalty_coef": 1.0, "quantile_encoding_window": 7, "quantile_smooth_coef": 0.8, "increment_loss_weight": 0.0, "covariate_training_mixing_coef": 1.0, "direction_loss_coef": 2000.0 }, encoder_specs=[ model_spec_lib.EncoderSpec( encoder_name="first_dose_vaccine_ratio_per_day", encoder_type="vaccine", vaccine_type="first_dose", covariate_feature_specs=[ first_dose_vaccine_ratio_per_day, first_dose_vaccine_effectiveness, second_dose_vaccine_effectiveness ], ), model_spec_lib.EncoderSpec( encoder_name="second_dose_vaccine_ratio_per_day", encoder_type="vaccine", vaccine_type="second_dose", covariate_feature_specs=[ second_dose_vaccine_ratio_per_day, first_dose_vaccine_effectiveness, second_dose_vaccine_effectiveness ], ), model_spec_lib.EncoderSpec( encoder_name="first_dose_vaccine_effectiveness", encoder_type="passthrough", covariate_feature_specs=[first_dose_vaccine_effectiveness], ), model_spec_lib.EncoderSpec( encoder_name="second_dose_vaccine_effectiveness", encoder_type="passthrough", covariate_feature_specs=[second_dose_vaccine_effectiveness], ), model_spec_lib.EncoderSpec(encoder_name="average_contact_id_rate", encoder_type="static"), model_spec_lib.EncoderSpec(encoder_name="average_contact_iud_rate", encoder_type="static"), model_spec_lib.EncoderSpec(encoder_name="reinfectable_rate", encoder_type="static"), model_spec_lib.EncoderSpec(encoder_name="alpha_rate", encoder_type="static"), model_spec_lib.EncoderSpec(encoder_name="diagnosis_rate", encoder_type="static"), model_spec_lib.EncoderSpec(encoder_name="recovery_id_rate", encoder_type="static"), model_spec_lib.EncoderSpec(encoder_name="recovery_iud_rate", encoder_type="static"), model_spec_lib.EncoderSpec(encoder_name="recovery_h_rate", encoder_type="static"), model_spec_lib.EncoderSpec(encoder_name="death_id_rate", encoder_type="static"), model_spec_lib.EncoderSpec(encoder_name="death_h_rate", encoder_type="static") ])
def get_model_spec_time_varying_with_covariates(covariate_delay): """Return model specification with hyperparameters and covariates.""" result = model_spec_lib.ModelSpec( hparams={ "initial_learning_rate": 0.003061242915556316, "momentum": 0.2, "decay_steps": 1000, "fine_tuning_steps": 100, "fine_tuning_decay": 1.0, "decay_rate": 1.0, "location_dependent_init": False, "infected_threshold": 3, "restart_threshold": 3000, "time_scale_weight": 0.00006243159539906051, "train_loss_coefs": [0.01, 0.01, 0.1, 0.3, 0.01, 0.01], "valid_loss_coefs": [0.01, 0.01, 0.1, 0.3, 0.01, 0.01], "sync_coef": 1.0, "reduced_sync_undoc": 1.0, "smooth_coef": 0.5, "first_dose_vaccine_ratio_per_day_init": 0.0, "second_dose_vaccine_ratio_per_day_init": 0.0, "average_contact_id_rate_init": -1.9131825459930378, "average_contact_iud_rate_init": -1.071866945725303, "reinfectable_rate_init": -5.548940468292865, "alpha_rate_init": -2.272765554778715, "diagnosis_rate_init": -2.095597433974376, "recovery_h_rate_init": -1.9032896753850963, "recovery_id_rate_init": -1.495660223962899, "recovery_iud_rate_init": -1.475605314236803, "hospitalization_rate_init": -1.4331763640928012, "death_h_rate_init": -1.4886876719378206, "death_id_rate_init": -1.8060447968974489, "bias_penalty_coef": 0.2835406167308398, "r_eff_penalty_coef": 2.0, "r_eff_penalty_cutoff": 5.0, "acceleration_death_coef": 0.1, "acceleration_confirm_coef": 0.1, "acceleration_hospital_coef": 0.1, "quantile_encoding_window": 7, "quantile_smooth_coef": 0.90, "quantile_training_iteration_ratio": 1.0, "width_coef_train": 1.0, "width_coef_valid": 1.0, "quantile_cum_viol_coef": 500.0, "increment_loss_weight": 0.0, "lasso_penalty_coef": 1.0, "covariate_training_mixing_coef": 1.0, "train_window_range": 2.0, "partial_mean_interval": 3, "direction_loss_coef": 2000.0, "train_crps_weight": 0.75, "valid_crps_weight": 0.75 }, encoder_specs=[ model_spec_lib.EncoderSpec( encoder_name="first_dose_vaccine_ratio_per_day", encoder_type="vaccine", vaccine_type="first_dose", covariate_feature_specs=[ first_dose_vaccine_ratio_per_day, first_dose_vaccine_effectiveness, second_dose_vaccine_effectiveness ], ), model_spec_lib.EncoderSpec( encoder_name="second_dose_vaccine_ratio_per_day", encoder_type="vaccine", vaccine_type="second_dose", covariate_feature_specs=[ second_dose_vaccine_ratio_per_day, first_dose_vaccine_effectiveness, second_dose_vaccine_effectiveness ], ), model_spec_lib.EncoderSpec( encoder_name="average_contact_id_rate", encoder_type="gam", static_feature_specs=demographics + wellness, covariate_feature_specs=[ confirmed_preprocessed, death_preprocessed, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, state_of_emergency, reproductive_number ] + survey_all + mobility_all, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="average_contact_iud_rate", encoder_type="gam", static_feature_specs=demographics + wellness, covariate_feature_specs=[ confirmed_preprocessed, death_preprocessed, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, state_of_emergency, reproductive_number ] + survey_all + mobility_all, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec(encoder_name="reinfectable_rate", encoder_type="gam", static_feature_specs=demographics + wellness, covariate_feature_specs=None, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, }), model_spec_lib.EncoderSpec(encoder_name="alpha_rate", encoder_type="gam", static_feature_specs=[], covariate_feature_specs=None, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, }), model_spec_lib.EncoderSpec( encoder_name="diagnosis_rate", encoder_type="gam", static_feature_specs=healthcare, covariate_feature_specs=[ confirmed_preprocessed, death_preprocessed, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, total_tests, dow_feature, ] + survey_all, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="recovery_id_rate", encoder_type="gam", static_feature_specs=demographics + wellness + healthcare, covariate_feature_specs=[ confirmed_preprocessed, death_preprocessed, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, ] + survey_all, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="recovery_iud_rate", encoder_type="gam", static_feature_specs=demographics + wellness + healthcare, covariate_feature_specs=[ confirmed_preprocessed, death_preprocessed, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, ] + survey_all, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="recovery_h_rate", encoder_type="gam", static_feature_specs=demographics + wellness + healthcare, covariate_feature_specs=[ confirmed_preprocessed, death_preprocessed, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, discharged, dow_feature, ], covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="hospitalization_rate", encoder_type="gam", static_feature_specs=demographics + wellness + healthcare, covariate_feature_specs=[ confirmed_preprocessed, death_preprocessed, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, discharged, dow_feature, ], covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="death_id_rate", encoder_type="gam", static_feature_specs=demographics + wellness + healthcare, covariate_feature_specs=[ confirmed_preprocessed, death_preprocessed, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, ], covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="death_h_rate", encoder_type="gam", static_feature_specs=demographics + wellness + healthcare, covariate_feature_specs=[ confirmed_preprocessed, death_preprocessed, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, discharged, dow_feature, ], covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), ]) return result
def get_model_spec_time_varying_with_covariates(covariate_delay): """Return model specification with hyperparameters and covariates.""" result = model_spec_lib.ModelSpec( hparams={ "initial_learning_rate": 0.003778327764151733, "momentum": 0.1, "decay_steps": 1000, "fine_tuning_steps": 100, "fine_tuning_decay": 1.0, "decay_rate": 1.0, "location_dependent_init": False, "infected_threshold": 10, "restart_threshold": 1000, "time_scale_weight": 0.00006243159539906051, "train_loss_coefs": [0, 0.001, 1.0, 0.1, 0.05, 0.05, 0.01, 0.001], "valid_loss_coefs": [0, 0.001, 1.0, 0.1, 0.05, 0.05, 0.01, 0.001], "sync_coef": 1.0, "reduced_sync_undoc": 1.0, "smooth_coef": 0.5, "first_dose_vaccine_ratio_per_day_init": 0.0, "second_dose_vaccine_ratio_per_day_init": 0.0, "average_contact_id_rate_init": -1.9131825459930378, "average_contact_iud_rate_init": -1.071866945725303, "reinfectable_rate_init": -5.548940468292865, "alpha_rate_init": -2.272765554778715, "diagnosis_rate_init": -2.095597433974376, "recovery_id_rate_init": -1.495660223962899, "recovery_iud_rate_init": -1.475605314236803, "recovery_h_rate_init": -1.9032896753850963, "recovery_i_rate_init": -1.354652366086861, "recovery_v_rate_init": -1.2872596880548643, "hospitalization_rate_init": -1.4331763640928012, "icu_rate_init": -1.7697853537123167, "ventilator_rate_init": -1.747746088542831, "death_id_rate_init": -1.8060447968974489, "death_h_rate_init": -1.4886876719378206, "death_i_rate_init": -1.525990911868795, "death_v_rate_init": -1.0072190496934017, "bias_penalty_coef": 0.2835406167308398, "r_eff_penalty_coef": 2.0, "r_eff_penalty_cutoff": 5.5, "acceleration_death_coef": 0.1, "acceleration_confirm_coef": 0.1, "acceleration_hospital_coef": 0.1, "quantile_encoding_window": 7, "quantile_smooth_coef": 0.9, "quantile_training_iteration_ratio": 1.0, "width_coef_train": 1.0, "width_coef_valid": 1.0, "quantile_cum_viol_coef": 1000.0, "increment_loss_weight": 0.0, "lasso_penalty_coef": 1.0, "covariate_training_mixing_coef": 1.0, "direction_loss_coef": 15000.0, "train_window_range": 2.0, "partial_mean_interval": 4, "train_crps_weight": 0.5, "valid_crps_weight": 0.5 }, encoder_specs=[ model_spec_lib.EncoderSpec( encoder_name="first_dose_vaccine_ratio_per_day", encoder_type="vaccine", vaccine_type="first_dose", covariate_feature_specs=[ first_dose_vaccine_ratio_per_day, first_dose_vaccine_effectiveness, second_dose_vaccine_effectiveness ], ), model_spec_lib.EncoderSpec( encoder_name="second_dose_vaccine_ratio_per_day", encoder_type="vaccine", vaccine_type="second_dose", covariate_feature_specs=[ second_dose_vaccine_ratio_per_day, first_dose_vaccine_effectiveness, second_dose_vaccine_effectiveness ], ), model_spec_lib.EncoderSpec( encoder_name="average_contact_id_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, income_per_capita, aqi_mean_2018, household, population_60plus_ratio ], covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, mobility_index, mobility_samples, average_temperature, rainfall, snowfall, antigen_positive_ratio, antibody_negative_ratio, confirmed_per_total_tests ] + npi_features, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="average_contact_iud_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, income_per_capita, aqi_mean_2018, household, population_60plus_ratio ], covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, mobility_index, mobility_samples, average_temperature, rainfall, snowfall, antigen_positive_ratio, antibody_negative_ratio, confirmed_per_total_tests ] + npi_features, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="reinfectable_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, income_per_capita, household_food_stamp, household, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, hospital_type_critical_access_hospitals, hospital_type_acute_care_department_of_defense, service_type_emergency_services_supported, service_type_non_emergency_services, rating_average, aqi_mean_2018, population_60plus_ratio ], covariate_feature_specs=None, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, }), model_spec_lib.EncoderSpec( encoder_name="alpha_rate", encoder_type="gam", static_feature_specs=[], covariate_feature_specs=None, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, }), model_spec_lib.EncoderSpec( encoder_name="diagnosis_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, income_per_capita, household, rating_average, population_60plus_ratio ], covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, total_tests_per_capita, confirmed_per_total_tests, dow_feature, antigen_positive_ratio, antibody_negative_ratio, average_temperature, rainfall, snowfall, ] + symptoms_all, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="recovery_id_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, income_per_capita, household_food_stamp, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, hospital_type_critical_access_hospitals, hospital_type_acute_care_department_of_defense, service_type_emergency_services_supported, service_type_non_emergency_services, rating_average, population_60plus_ratio ], covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, antigen_positive_ratio, antibody_negative_ratio, average_temperature, rainfall, snowfall, ], covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="recovery_iud_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, income_per_capita, household_food_stamp, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, hospital_type_critical_access_hospitals, hospital_type_acute_care_department_of_defense, service_type_emergency_services_supported, service_type_non_emergency_services, rating_average, population_60plus_ratio ], covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, antigen_positive_ratio, antibody_negative_ratio, average_temperature, rainfall, snowfall, ], covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="recovery_h_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, income_per_capita, household_food_stamp, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, hospital_type_critical_access_hospitals, hospital_type_acute_care_department_of_defense, service_type_emergency_services_supported, service_type_non_emergency_services, rating_average, population_60plus_ratio ], covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, ], covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="recovery_i_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, income_per_capita, household_food_stamp, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, hospital_type_critical_access_hospitals, hospital_type_acute_care_department_of_defense, service_type_emergency_services_supported, service_type_non_emergency_services, rating_average, population_60plus_ratio ], covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, ], covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="recovery_v_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, income_per_capita, household_food_stamp, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, hospital_type_critical_access_hospitals, hospital_type_acute_care_department_of_defense, service_type_emergency_services_supported, service_type_non_emergency_services, rating_average, population_60plus_ratio ], covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, ], covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="hospitalization_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, household_food_stamp, household, income_per_capita, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, hospital_type_critical_access_hospitals, hospital_type_acute_care_department_of_defense, service_type_emergency_services_supported, service_type_non_emergency_services, rating_average, aqi_mean_2018, icu_beds, population_60plus_ratio, ], covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, total_tests_per_capita, confirmed_per_total_tests, antigen_positive_ratio, antibody_negative_ratio, average_temperature, rainfall, snowfall ] + symptoms_all, covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="icu_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, household_food_stamp, household, income_per_capita, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, hospital_type_critical_access_hospitals, hospital_type_acute_care_department_of_defense, service_type_emergency_services_supported, service_type_non_emergency_services, rating_average, aqi_mean_2018, icu_beds, population_60plus_ratio ], covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, ], covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="ventilator_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, household_food_stamp, household, income_per_capita, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, hospital_type_critical_access_hospitals, hospital_type_acute_care_department_of_defense, service_type_emergency_services_supported, service_type_non_emergency_services, rating_average, aqi_mean_2018, icu_beds, population_60plus_ratio ], covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, ], covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="death_id_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, household_food_stamp, income_per_capita, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, hospital_type_critical_access_hospitals, hospital_type_acute_care_department_of_defense, service_type_emergency_services_supported, service_type_non_emergency_services, rating_average, aqi_mean_2018, icu_beds, population_60plus_ratio ], covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, ], covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="death_h_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, household_food_stamp, income_per_capita, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, hospital_type_critical_access_hospitals, hospital_type_acute_care_department_of_defense, service_type_emergency_services_supported, service_type_non_emergency_services, rating_average, aqi_mean_2018, icu_beds, population_60plus_ratio ], covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, ], covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="death_i_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, household_food_stamp, income_per_capita, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, hospital_type_critical_access_hospitals, hospital_type_acute_care_department_of_defense, service_type_emergency_services_supported, service_type_non_emergency_services, rating_average, aqi_mean_2018, icu_beds, population_60plus_ratio ], covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, ], covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), model_spec_lib.EncoderSpec( encoder_name="death_v_rate", encoder_type="gam", static_feature_specs=[ population_density_per_sq_km, household, household_food_stamp, income_per_capita, patient_experience_above_the_national_average, patient_experience_below_the_national_average, patient_experience_same_as_the_national_average, hospital_type_critical_access_hospitals, hospital_type_acute_care_department_of_defense, service_type_emergency_services_supported, service_type_non_emergency_services, rating_average, aqi_mean_2018, icu_beds, population_60plus_ratio ], covariate_feature_specs=[ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, dow_feature, ], covariate_feature_time_offset=covariate_delay, covariate_feature_window=DEFAULT_WINDOW_SIZE, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": USE_FIXED_COVARIATE_MASK, }), ]) return result
STATIC_MODEL = model_spec_lib.ModelSpec( hparams={ "initial_learning_rate": 0.003046320146869829, "decay_steps": 5, "decay_rate": 1.0, "fine_tuning_steps": 1000, "location_dependent_init": True, "infected_threshold": 10, "restart_threshold": 300, "time_scale_weight": 0.00007493264308148026, "train_loss_coefs": [0, 0, 1, 0.001131, 1e-7, 1, 0.0045], "valid_loss_coefs": [0, 0, 1, 0.0, 0.0, 0.0, 0.0], "sync_coef": 0.5, "first_dose_vaccine_ratio_per_day_init": 0.0, "second_dose_vaccine_ratio_per_day_init": 0.0, "average_contact_id_rate_init": -2.4387148524945594, "average_contact_iud_rate_init": -1.561117964142531, "reinfectable_rate_init": -5.447821316098564, "alpha_rate_init": -1.1906569150841058, "diagnosis_rate_init": -1.589940307765024, "recovery_id_rate_init": -1.5723201455595701, "recovery_iud_rate_init": -1.8295556397198884, "recovery_h_rate_init": -1.828568757553072, "recovery_i_rate_init": -1.5923711129624136, "recovery_v_rate_init": -1.7109958618683396, "hospitalization_rate_init": -1.5327844461092095, "icu_rate_init": -1.6822889372036114, "ventilator_rate_init": -1.6721172663784372, "death_id_rate_init": -1.0766994750650696, "death_h_rate_init": -1.3775460164801863, "death_i_rate_init": -1.8531814769372583, "death_v_rate_init": -1.3251172286148205, "compartmental_penalty_coef": 96.12757205042173, "variable_smoothness_penalty_coef": 5.988152939976956, "bias_penalty_coef": 0.27315803186400106, "boundary_penalty_coef": 2.600265396382176, "lasso_penalty_coef": 1.0, "quantile_encoding_window": 7, "quantile_smooth_coef": 0.8, "width_coef": 5.0, "increment_loss_weight": 0.0, "covariate_training_mixing_coef": 1.0, "direction_loss_coef": 20000.0 }, encoder_specs=[ model_spec_lib.EncoderSpec( encoder_name="first_dose_vaccine_ratio_per_day", encoder_type="vaccine", vaccine_type="first_dose", covariate_feature_specs=[ first_dose_vaccine_ratio_per_day, first_dose_vaccine_effectiveness, second_dose_vaccine_effectiveness ], ), model_spec_lib.EncoderSpec( encoder_name="second_dose_vaccine_ratio_per_day", encoder_type="vaccine", vaccine_type="second_dose", covariate_feature_specs=[ second_dose_vaccine_ratio_per_day, first_dose_vaccine_effectiveness, second_dose_vaccine_effectiveness ], ), model_spec_lib.EncoderSpec( encoder_name="average_contact_id_rate", encoder_type="static"), model_spec_lib.EncoderSpec( encoder_name="average_contact_iud_rate", encoder_type="static"), model_spec_lib.EncoderSpec( encoder_name="reinfectable_rate", encoder_type="static"), model_spec_lib.EncoderSpec( encoder_name="alpha_rate", encoder_type="static"), model_spec_lib.EncoderSpec( encoder_name="diagnosis_rate", encoder_type="static"), model_spec_lib.EncoderSpec( encoder_name="recovery_id_rate", encoder_type="static"), model_spec_lib.EncoderSpec( encoder_name="recovery_iud_rate", encoder_type="static"), model_spec_lib.EncoderSpec( encoder_name="recovery_h_rate", encoder_type="static"), model_spec_lib.EncoderSpec( encoder_name="recovery_i_rate", encoder_type="static"), model_spec_lib.EncoderSpec( encoder_name="recovery_v_rate", encoder_type="static"), model_spec_lib.EncoderSpec( encoder_name="hospitalization_rate", encoder_type="static"), model_spec_lib.EncoderSpec( encoder_name="icu_rate", encoder_type="static"), model_spec_lib.EncoderSpec( encoder_name="ventilator_rate", encoder_type="static"), model_spec_lib.EncoderSpec( encoder_name="death_id_rate", encoder_type="static"), model_spec_lib.EncoderSpec( encoder_name="death_h_rate", encoder_type="static"), model_spec_lib.EncoderSpec( encoder_name="death_i_rate", encoder_type="static"), model_spec_lib.EncoderSpec( encoder_name="death_v_rate", encoder_type="static") ])
def get_model_spec( self, model_type, covariate_delay=0, **kwargs, ): # Static feature candidates population_density_per_sq_km = model_spec_lib.FeatureSpec( name=constants.POPULATION_DENSITY_PER_SQKM, initializer=None) # Time-varying feature candidates preprocessed_confirmed = model_spec_lib.FeatureSpec( name=constants.CONFIRMED_PREPROCESSED, initializer=None, forecast_method=model_spec_lib.ForecastMethod.NONE) preprocessed_death = model_spec_lib.FeatureSpec( name=constants.DEATH_PREPROCESSED, initializer=None, forecast_method=model_spec_lib.ForecastMethod.NONE) preprocessed_confirmed_mean_to_sum = model_spec_lib.FeatureSpec( name=constants.CONFIRMED_PREPROCESSED_MEAN_TO_SUM_RATIO, initializer=None, forecast_method=model_spec_lib.ForecastMethod.NONE) preprocessed_death_mean_to_sum = model_spec_lib.FeatureSpec( name=constants.DEATH_PREPROCESSED_MEAN_TO_SUM_RATIO, initializer=None, forecast_method=model_spec_lib.ForecastMethod.NONE) first_dose_vaccine_ratio_per_day = model_spec_lib.FeatureSpec( name=constants.VACCINATED_RATIO_FIRST_DOSE_PER_DAY_PREPROCESSED, initializer=None) second_dose_vaccine_ratio_per_day = model_spec_lib.FeatureSpec( name=constants.VACCINATED_RATIO_SECOND_DOSE_PER_DAY_PREPROCESSED, initializer=None) first_dose_vaccine_effectiveness = model_spec_lib.FeatureSpec( name=constants.VACCINATED_EFFECTIVENESS_FIRST_DOSE, initializer=None) second_dose_vaccine_effectiveness = model_spec_lib.FeatureSpec( name=constants.VACCINATED_EFFECTIVENESS_SECOND_DOSE, initializer=None) mobility_index = model_spec_lib.FeatureSpec( name=constants.MOBILITY_INDEX, initializer=None) all_static_covariates = [population_density_per_sq_km] base_ts_covariates = [ preprocessed_confirmed, preprocessed_death, preprocessed_confirmed_mean_to_sum, preprocessed_death_mean_to_sum, ] default_window_size = 7 use_fixed_covariate_mask = True return model_spec_lib.ModelSpec( hparams={ "initial_learning_rate": 0.003061242915556316, "momentum": 0.2, "decay_steps": 1000, "fine_tuning_steps": 500, "fine_tuning_decay": 1.0, "decay_rate": 1.0, "location_dependent_init": False, "infected_threshold": 3, "restart_threshold": 300, "time_scale_weight": 0.00006243159539906051, "train_loss_coefs": [0, 0.001, 0.2, 0.1, 0.01, 0.01], "valid_loss_coefs": [0, 0.001, 0.2, 0.1, 0.01, 0.01], "sync_coef": 0.3, "reduced_sync_undoc": 2.0, "smooth_coef": 0.5, "first_dose_vaccine_ratio_per_day_init": 0.0, "second_dose_vaccine_ratio_per_day_init": 0.0, "average_contact_id_rate_init": -1.9131825459930378, "average_contact_iud_rate_init": -1.071866945725303, "reinfectable_rate_init": -5.548940468292865, "alpha_rate_init": -2.272765554778715, "diagnosis_rate_init": -2.095597433974376, "recovery_id_rate_init": -1.495660223962899, "recovery_iud_rate_init": -1.475605314236803, "recovery_h_rate_init": -1.9032896753850963, "hospitalization_rate_init": -1.4331763640928012, "death_id_rate_init": -1.8060447968974489, "death_h_rate_init": -1.4886876719378206, "bias_penalty_coef": 0.2835406167308398, "r_eff_penalty_coef": 2.0, "acceleration_death_coef": 0.1, "acceleration_confirm_coef": 0.1, "acceleration_hospital_coef": 0.1, "quantile_encoding_window": 7, "quantile_smooth_coef": 0.9, "quantile_training_iteration_ratio": 0.5, "width_coef_train": 10.0, "width_coef_valid": 5.0, "quantile_cum_viol_coef": 500.0, "increment_loss_weight": 0.0, "lasso_penalty_coef": 1.0, "covariate_training_mixing_coef": 1.0, "train_window_range": 2.0, "direction_loss_coef": 5000.0, "train_crps_weight": 0.25, "valid_crps_weight": 0.25 }, encoder_specs=[ model_spec_lib.EncoderSpec( encoder_name="first_dose_vaccine_ratio_per_day", encoder_type="vaccine", vaccine_type="first_dose", covariate_feature_specs=[ first_dose_vaccine_ratio_per_day, first_dose_vaccine_effectiveness, second_dose_vaccine_effectiveness ], ), model_spec_lib.EncoderSpec( encoder_name="second_dose_vaccine_ratio_per_day", encoder_type="vaccine", vaccine_type="second_dose", covariate_feature_specs=[ second_dose_vaccine_ratio_per_day, first_dose_vaccine_effectiveness, second_dose_vaccine_effectiveness ], ), model_spec_lib.EncoderSpec( encoder_name="average_contact_id_rate", encoder_type="gam", static_feature_specs=all_static_covariates, covariate_feature_specs=base_ts_covariates + [mobility_index], covariate_feature_window=default_window_size, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": use_fixed_covariate_mask, }), model_spec_lib.EncoderSpec( encoder_name="average_contact_iud_rate", encoder_type="gam", static_feature_specs=all_static_covariates, covariate_feature_specs=base_ts_covariates + [mobility_index], covariate_feature_window=default_window_size, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": use_fixed_covariate_mask, }), model_spec_lib.EncoderSpec( encoder_name="reinfectable_rate", encoder_type="gam", static_feature_specs=all_static_covariates, covariate_feature_specs=None, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, }), model_spec_lib.EncoderSpec(encoder_name="alpha_rate", encoder_type="gam", static_feature_specs=[], covariate_feature_specs=None, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, }), model_spec_lib.EncoderSpec( encoder_name="diagnosis_rate", encoder_type="gam", static_feature_specs=all_static_covariates, covariate_feature_specs=base_ts_covariates, covariate_feature_window=default_window_size, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": use_fixed_covariate_mask, }), model_spec_lib.EncoderSpec( encoder_name="recovery_id_rate", encoder_type="gam", static_feature_specs=all_static_covariates, covariate_feature_specs=base_ts_covariates, covariate_feature_window=default_window_size, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": use_fixed_covariate_mask, }), model_spec_lib.EncoderSpec( encoder_name="recovery_iud_rate", encoder_type="gam", static_feature_specs=all_static_covariates, covariate_feature_specs=base_ts_covariates, covariate_feature_window=default_window_size, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": use_fixed_covariate_mask, }), model_spec_lib.EncoderSpec( encoder_name="recovery_h_rate", encoder_type="gam", static_feature_specs=all_static_covariates, covariate_feature_specs=base_ts_covariates, covariate_feature_window=default_window_size, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": use_fixed_covariate_mask, }), model_spec_lib.EncoderSpec( encoder_name="hospitalization_rate", encoder_type="gam", static_feature_specs=all_static_covariates, covariate_feature_specs=base_ts_covariates, covariate_feature_window=default_window_size, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": use_fixed_covariate_mask, }), model_spec_lib.EncoderSpec( encoder_name="death_id_rate", encoder_type="gam", static_feature_specs=all_static_covariates, covariate_feature_specs=base_ts_covariates, covariate_feature_window=default_window_size, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": use_fixed_covariate_mask, }), model_spec_lib.EncoderSpec( encoder_name="death_h_rate", encoder_type="gam", static_feature_specs=all_static_covariates, covariate_feature_specs=base_ts_covariates, covariate_feature_window=default_window_size, encoder_kwargs={ "initial_bias": 0, "location_dependent_bias": True, "use_fixed_covariate_mask": use_fixed_covariate_mask, }), ])