Skip to content

Commit

Permalink
added new experiments
Browse files Browse the repository at this point in the history
  • Loading branch information
jguski committed May 13, 2024
1 parent 65bfea0 commit 57e0b03
Show file tree
Hide file tree
Showing 120 changed files with 8,346 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,66 @@
# @package _global_
defaults:
- /model/causal_model/model_config: crnlearner_DE_nuts0
- override /model/causal_model: crnlearner
- override /model/inference: variational_dropout
- override /refutation:
- propensity_thresholding_test
- placebo_treatment_refutation
- random_common_cause_refutation
- data_subset_refutation

general:
experiment_name: nuts0_geo_flag/DE
input_data: ./data/data.csv
out_dir_base: ./results
substeps:
- fit_causal_model
- evaluate_causal_model
- refutation
#- shap_analysis
data:
treatment: npi_internal_travel_filtered
treatment_levels: 2
outcome: r_number
outcome_lag: 7
time_predictors:
- r_number
- mob_transit
- weather_iptcc
- weather_mean_temperature
- weather_humidity
- weather_precipitation
- pollution_o3
- pollution_no2
- pollution_pm10
- covid_tests
- covid_mask_use
- covid_vacc_policy
- npi_stay_home
- npi_work
- npi_schools
- npi_international_travel
countries:
- DE
regions:
- DE
period_fit:
- 2020-11
- 2022-22
period_eval:
- 2022-22
- 2022-35
validation_start: 2022-17
window_size: ${model.causal_model.model_config.window_size}
model: # set baselearners to null for CRN
y_models:
model_class: null
t_models:
model_class: null
cate_models:
model_class: null
# shap:
# defined_periods:
# - [2020-12,2020-22]
# - [2020-40,2021-08]
# - [2022-04,2022-34]
Original file line number Diff line number Diff line change
@@ -0,0 +1,66 @@
# @package _global_
defaults:
- /model/causal_model/model_config: crnlearner_ES_nuts0
- override /model/causal_model: crnlearner
- override /model/inference: variational_dropout
- override /refutation:
- propensity_thresholding_test
- placebo_treatment_refutation
- random_common_cause_refutation
- data_subset_refutation

general:
experiment_name: nuts0_geo_flag/ES
input_data: ./data/data.csv
out_dir_base: ./results
substeps:
- fit_causal_model
- evaluate_causal_model
- refutation
#- shap_analysis
data:
treatment: npi_internal_travel_filtered
treatment_levels: 2
outcome: r_number
outcome_lag: 7
time_predictors:
- r_number
- mob_transit
- weather_iptcc
- weather_mean_temperature
- weather_humidity
- weather_precipitation
- pollution_o3
- pollution_no2
- pollution_pm10
- covid_tests
- covid_mask_use
- covid_vacc_policy
- npi_stay_home
- npi_work
- npi_schools
- npi_international_travel
countries:
- ES
regions:
- ES
period_fit:
- 2020-7
- 2021-51
period_eval:
- 2021-51
- 2022-12
validation_start: 2021-47
window_size: ${model.causal_model.model_config.window_size}
model: # set baselearners to null for CRN
y_models:
model_class: null
t_models:
model_class: null
cate_models:
model_class: null
# shap:
# defined_periods:
# - [2020-12,2020-22]
# - [2020-40,2021-08]
# - [2022-04,2022-34]
Original file line number Diff line number Diff line change
@@ -0,0 +1,66 @@
# @package _global_
defaults:
- /model/causal_model/model_config: crnlearner_FR_nuts0
- override /model/causal_model: crnlearner
- override /model/inference: variational_dropout
- override /refutation:
- propensity_thresholding_test
- placebo_treatment_refutation
- random_common_cause_refutation
- data_subset_refutation

general:
experiment_name: nuts0_geo_flag/FR
input_data: ./data/data.csv
out_dir_base: ./results
substeps:
- fit_causal_model
- evaluate_causal_model
- refutation
#- shap_analysis
data:
treatment: npi_internal_travel_filtered
treatment_levels: 2
outcome: r_number
outcome_lag: 7
time_predictors:
- r_number
- mob_transit
- weather_iptcc
- weather_mean_temperature
- weather_humidity
- weather_precipitation
- pollution_o3
- pollution_no2
- pollution_pm10
- covid_tests
- covid_mask_use
- covid_vacc_policy
- npi_stay_home
- npi_work
- npi_schools
- npi_international_travel
countries:
- FR
regions:
- FR
period_fit:
- 2020-12
- 2022-21
period_eval:
- 2022-21
- 2022-34
validation_start: 2022-17
window_size: ${model.causal_model.model_config.window_size}
model: # set baselearners to null for CRN
y_models:
model_class: null
t_models:
model_class: null
cate_models:
model_class: null
# shap:
# defined_periods:
# - [2020-12,2020-22]
# - [2020-40,2021-08]
# - [2022-04,2022-34]
Original file line number Diff line number Diff line change
@@ -0,0 +1,66 @@
# @package _global_
defaults:
- /model/causal_model/model_config: crnlearner_BE_nuts0
- override /model/causal_model: crnlearner
- override /model/inference: variational_dropout
- override /refutation:
- propensity_thresholding_test
- placebo_treatment_refutation
- random_common_cause_refutation
- data_subset_refutation

general:
experiment_name: nuts0_geo_flag/BE
input_data: ./data/data.csv
out_dir_base: ./results
substeps:
- fit_causal_model
- evaluate_causal_model
- refutation
#- shap_analysis
data:
treatment: npi_international_travel_filtered
treatment_levels: 2
outcome: r_number
outcome_lag: 7
time_predictors:
- r_number
- mob_transit
- weather_iptcc
- weather_mean_temperature
- weather_humidity
- weather_precipitation
- pollution_o3
- pollution_no2
- pollution_pm10
- covid_tests
- covid_mask_use
- covid_vacc_policy
- npi_stay_home
- npi_work
- npi_internal_travel
- npi_schools
countries:
- BE
regions:
- BE
period_fit:
- 2020-11
- 2022-21
period_eval:
- 2022-21
- 2022-34
validation_start: 2022-17
window_size: ${model.causal_model.model_config.window_size}
model: # set baselearners to null for CRN
y_models:
model_class: null
t_models:
model_class: null
cate_models:
model_class: null
# shap:
# defined_periods:
# - [2020-12,2020-22]
# - [2020-40,2021-08]
# - [2022-04,2022-34]
Original file line number Diff line number Diff line change
@@ -0,0 +1,65 @@
# @package _global_
defaults:
- /model/causal_model/model_config: crnlearner_DE_nuts0
- override /model/causal_model: crnlearner
- override /model/inference: variational_dropout
- override /refutation:
- propensity_thresholding_test
- placebo_treatment_refutation
- random_common_cause_refutation
- data_subset_refutation

general:
experiment_name: nuts0_geo_flag/DE
input_data: ./data/data.csv
out_dir_base: ./results
substeps:
- fit_causal_model
- evaluate_causal_model
- refutation
#- shap_analysis
data:
treatment: npi_international_travel_filtered
treatment_levels: 2
outcome: r_number
outcome_lag: 7
time_predictors:
- r_number
- mob_transit
- weather_iptcc
- weather_mean_temperature
- weather_humidity
- weather_precipitation
- pollution_o3
- pollution_no2
- pollution_pm10
- covid_tests
- covid_mask_use
- covid_vacc_policy
- npi_stay_home
- npi_work
- npi_internal_travel
- npi_schools
countries:
- DE
regions:
- DE
period_fit:
- 2020-11
- 2022-22
period_eval:
- 2022-22
- 2022-35
validation_start: 2022-17
window_size: ${model.causal_model.model_config.window_size}
model: # set baselearners to null for CRN
y_models:
model_class: null
t_models:
model_class: null
cate_models:
model_class: null
shap:
defined_periods:
#- [2022-22,2022-35]
- [2020-11,2022-22]
Loading

0 comments on commit 57e0b03

Please sign in to comment.