.gitignore
.pre-commit-config.yaml
LICENSE
README.md
pyproject.toml
.github/workflows/python-build.yml
.github/workflows/tests.yml
sample_data/omop_sample/concept/._SUCCESS.crc
sample_data/omop_sample/concept/.part-00000-4b12270c-f6c8-4b59-8e0f-fd588bd79386-c000.snappy.parquet.crc
sample_data/omop_sample/concept/.part-00003-4b12270c-f6c8-4b59-8e0f-fd588bd79386-c000.snappy.parquet.crc
sample_data/omop_sample/concept/.part-00010-4b12270c-f6c8-4b59-8e0f-fd588bd79386-c000.snappy.parquet.crc
sample_data/omop_sample/concept/_SUCCESS
sample_data/omop_sample/concept/part-00000-4b12270c-f6c8-4b59-8e0f-fd588bd79386-c000.snappy.parquet
sample_data/omop_sample/concept/part-00003-4b12270c-f6c8-4b59-8e0f-fd588bd79386-c000.snappy.parquet
sample_data/omop_sample/concept/part-00010-4b12270c-f6c8-4b59-8e0f-fd588bd79386-c000.snappy.parquet
sample_data/omop_sample/concept_ancestor/._SUCCESS.crc
sample_data/omop_sample/concept_ancestor/.part-00000-eafbd8be-3337-46da-89d3-20f79c2565d4-c000.snappy.parquet.crc
sample_data/omop_sample/concept_ancestor/.part-00002-eafbd8be-3337-46da-89d3-20f79c2565d4-c000.snappy.parquet.crc
sample_data/omop_sample/concept_ancestor/.part-00006-eafbd8be-3337-46da-89d3-20f79c2565d4-c000.snappy.parquet.crc
sample_data/omop_sample/concept_ancestor/.part-00011-eafbd8be-3337-46da-89d3-20f79c2565d4-c000.snappy.parquet.crc
sample_data/omop_sample/concept_ancestor/.part-00013-eafbd8be-3337-46da-89d3-20f79c2565d4-c000.snappy.parquet.crc
sample_data/omop_sample/concept_ancestor/_SUCCESS
sample_data/omop_sample/concept_ancestor/part-00000-eafbd8be-3337-46da-89d3-20f79c2565d4-c000.snappy.parquet
sample_data/omop_sample/concept_ancestor/part-00002-eafbd8be-3337-46da-89d3-20f79c2565d4-c000.snappy.parquet
sample_data/omop_sample/concept_ancestor/part-00006-eafbd8be-3337-46da-89d3-20f79c2565d4-c000.snappy.parquet
sample_data/omop_sample/concept_ancestor/part-00011-eafbd8be-3337-46da-89d3-20f79c2565d4-c000.snappy.parquet
sample_data/omop_sample/concept_ancestor/part-00013-eafbd8be-3337-46da-89d3-20f79c2565d4-c000.snappy.parquet
sample_data/omop_sample/concept_relationship/._SUCCESS.crc
sample_data/omop_sample/concept_relationship/.part-00000-5752b472-8ba7-4189-ab69-8c92e46443aa-c000.snappy.parquet.crc
sample_data/omop_sample/concept_relationship/.part-00002-5752b472-8ba7-4189-ab69-8c92e46443aa-c000.snappy.parquet.crc
sample_data/omop_sample/concept_relationship/.part-00007-5752b472-8ba7-4189-ab69-8c92e46443aa-c000.snappy.parquet.crc
sample_data/omop_sample/concept_relationship/.part-00012-5752b472-8ba7-4189-ab69-8c92e46443aa-c000.snappy.parquet.crc
sample_data/omop_sample/concept_relationship/_SUCCESS
sample_data/omop_sample/concept_relationship/part-00000-5752b472-8ba7-4189-ab69-8c92e46443aa-c000.snappy.parquet
sample_data/omop_sample/concept_relationship/part-00002-5752b472-8ba7-4189-ab69-8c92e46443aa-c000.snappy.parquet
sample_data/omop_sample/concept_relationship/part-00007-5752b472-8ba7-4189-ab69-8c92e46443aa-c000.snappy.parquet
sample_data/omop_sample/concept_relationship/part-00012-5752b472-8ba7-4189-ab69-8c92e46443aa-c000.snappy.parquet
sample_data/omop_sample/condition_occurrence/._SUCCESS.crc
sample_data/omop_sample/condition_occurrence/.part-00000-4eff03a1-cdcf-4c89-b0cd-9ce590b9b1eb-c000.snappy.parquet.crc
sample_data/omop_sample/condition_occurrence/_SUCCESS
sample_data/omop_sample/condition_occurrence/part-00000-4eff03a1-cdcf-4c89-b0cd-9ce590b9b1eb-c000.snappy.parquet
sample_data/omop_sample/drug_exposure/._SUCCESS.crc
sample_data/omop_sample/drug_exposure/.part-00000-10bbf1a4-a7da-416e-9703-58609c7edfad-c000.snappy.parquet.crc
sample_data/omop_sample/drug_exposure/_SUCCESS
sample_data/omop_sample/drug_exposure/part-00000-10bbf1a4-a7da-416e-9703-58609c7edfad-c000.snappy.parquet
sample_data/omop_sample/observation_period/._SUCCESS.crc
sample_data/omop_sample/observation_period/.part-00000-694316e5-cc95-49f1-9fad-5a7f377e2602-c000.snappy.parquet.crc
sample_data/omop_sample/observation_period/_SUCCESS
sample_data/omop_sample/observation_period/part-00000-694316e5-cc95-49f1-9fad-5a7f377e2602-c000.snappy.parquet
sample_data/omop_sample/person/._SUCCESS.crc
sample_data/omop_sample/person/.part-00000-7d789011-f361-48da-af6f-cfe102978b3a-c000.snappy.parquet.crc
sample_data/omop_sample/person/_SUCCESS
sample_data/omop_sample/person/part-00000-7d789011-f361-48da-af6f-cfe102978b3a-c000.snappy.parquet
sample_data/omop_sample/procedure_occurrence/._SUCCESS.crc
sample_data/omop_sample/procedure_occurrence/.part-00000-e73003c1-aed5-41c0-b2d4-eaccaccf044a-c000.snappy.parquet.crc
sample_data/omop_sample/procedure_occurrence/_SUCCESS
sample_data/omop_sample/procedure_occurrence/part-00000-e73003c1-aed5-41c0-b2d4-eaccaccf044a-c000.snappy.parquet
sample_data/omop_sample/visit_occurrence/._SUCCESS.crc
sample_data/omop_sample/visit_occurrence/.part-00000-e874b5f1-bf9e-4cb9-93bf-c309a47b0476-c000.snappy.parquet.crc
sample_data/omop_sample/visit_occurrence/_SUCCESS
sample_data/omop_sample/visit_occurrence/part-00000-e874b5f1-bf9e-4cb9-93bf-c309a47b0476-c000.snappy.parquet
scripts/extract_features_bert.sh
scripts/extract_features_gpt.sh
scripts/process_cohorts.sh
src/__init__.py
src/cehrbert_data/__init__.py
src/cehrbert_data.egg-info/PKG-INFO
src/cehrbert_data.egg-info/SOURCES.txt
src/cehrbert_data.egg-info/dependency_links.txt
src/cehrbert_data.egg-info/requires.txt
src/cehrbert_data.egg-info/top_level.txt
src/cehrbert_data/apps/__init__.py
src/cehrbert_data/apps/generate_included_concept_list.py
src/cehrbert_data/apps/generate_training_data.py
src/cehrbert_data/cohorts/__init__.py
src/cehrbert_data/cohorts/atrial_fibrillation.py
src/cehrbert_data/cohorts/cabg.py
src/cehrbert_data/cohorts/coronary_artery_disease.py
src/cehrbert_data/cohorts/covid.py
src/cehrbert_data/cohorts/covid_inpatient.py
src/cehrbert_data/cohorts/death.py
src/cehrbert_data/cohorts/heart_failure.py
src/cehrbert_data/cohorts/ischemic_stroke.py
src/cehrbert_data/cohorts/last_visit_discharged_home.py
src/cehrbert_data/cohorts/query_builder.py
src/cehrbert_data/cohorts/spark_app_base.py
src/cehrbert_data/cohorts/type_two_diabietes.py
src/cehrbert_data/cohorts/ventilation.py
src/cehrbert_data/config/__init__.py
src/cehrbert_data/config/output_names.py
src/cehrbert_data/const/__init__.py
src/cehrbert_data/const/artificial_tokens.py
src/cehrbert_data/const/common.py
src/cehrbert_data/const/__pycache__/__init__.cpython-311.pyc
src/cehrbert_data/const/__pycache__/common.cpython-311.pyc
src/cehrbert_data/decorators/__init__.py
src/cehrbert_data/decorators/artificial_time_token_decorator.py
src/cehrbert_data/decorators/clinical_event_decorator.py
src/cehrbert_data/decorators/death_event_decorator.py
src/cehrbert_data/decorators/demographic_event_decorator.py
src/cehrbert_data/decorators/patient_event_decorator_base.py
src/cehrbert_data/decorators/prediction_token_decorator.py
src/cehrbert_data/decorators/token_priority.py
src/cehrbert_data/prediction_cohorts/__init__.py
src/cehrbert_data/prediction_cohorts/afib_ischemic_stroke.py
src/cehrbert_data/prediction_cohorts/cad_cabg_cohort.py
src/cehrbert_data/prediction_cohorts/cad_hf_cohort.py
src/cehrbert_data/prediction_cohorts/copd_readmission.py
src/cehrbert_data/prediction_cohorts/covid_death.py
src/cehrbert_data/prediction_cohorts/covid_ventilation.py
src/cehrbert_data/prediction_cohorts/discharge_home_death.py
src/cehrbert_data/prediction_cohorts/hf_readmission.py
src/cehrbert_data/prediction_cohorts/hospitalization.py
src/cehrbert_data/prediction_cohorts/hospitalization_mortality.py
src/cehrbert_data/prediction_cohorts/readmission.py
src/cehrbert_data/prediction_cohorts/t2dm_hf_cohort.py
src/cehrbert_data/queries/__init__.py
src/cehrbert_data/queries/measurement_queries.py
src/cehrbert_data/tools/__init__.py
src/cehrbert_data/tools/connect_omop_visit.py
src/cehrbert_data/tools/convert_prediction_time_to_local.py
src/cehrbert_data/tools/convert_prediction_time_to_str.py
src/cehrbert_data/tools/download_omop_tables.py
src/cehrbert_data/tools/ehrshot_to_omop.py
src/cehrbert_data/tools/extract_features.py
src/cehrbert_data/tools/prepare_ehrshot_cohorts.py
src/cehrbert_data/tools/sample_omop_tables.py
src/cehrbert_data/tools/update_omop_visit.py
src/cehrbert_data/utils/__init__.py
src/cehrbert_data/utils/logging_utils.py
src/cehrbert_data/utils/spark_parse_args.py
src/cehrbert_data/utils/spark_utils.py
src/cehrbert_data/utils/vocab_utils.py
tests/__init__.py
tests/pyspark_test_base.py
tests/integration_tests/__init__.py
tests/integration_tests/test_generate_training_data.py
tests/integration_tests/test_hf_readmission.py
tests/integration_tests/test_hf_readmission_cohort_meds.py
tests/unit_tests/__init__.py
tests/unit_tests/test_ehrshot_to_omop.py
tests/unit_tests/test_spark_utils.py