| from utils.adm_common import (initialize_adm_data, correct_stays, | |
| track_copd_resp) | |
| def main(): | |
| # Load in data | |
| adm_file = "<YOUR_DATA_PATH>/EXAMPLE_STUDY_DATA/SMR01_Cohort3R.csv" | |
| adm = initialize_adm_data(adm_file) | |
| # Fill null STAY data and combine transfer admissions | |
| adm = correct_stays(adm) | |
| # Track COPD and respiratory events | |
| adm = track_copd_resp(adm) | |
| # Select relevant columns | |
| adm_reduced = adm[['SafeHavenID', 'ADMDATE', 'copd_event', 'resp_event']] | |
| # Track events | |
| adm_reduced['copd_resp_event'] = adm_reduced['copd_event'] | adm_reduced['resp_event'] | |
| adm_reduced['adm_event'] = 1 | |
| # Save data | |
| adm_reduced.to_pickle('<YOUR_DATA_PATH>/Model_E_Extracts/validation_adm_proc-og.pkl') | |
| main() | |