├── .gitignore ├── LICENSE ├── PandemicLLM.jpg ├── Readme.md ├── configs ├── data │ └── covid.yaml ├── exp │ └── sft.yaml ├── llm │ ├── llama_peft.yaml │ └── llm_meta_data.yaml ├── main.yaml ├── model │ └── covid_llm.yaml └── prompt │ └── prompts.yaml ├── data ├── Anonymized.pkl ├── data_v6_after_2021-06-01.pkl ├── data_v6_after_2021-12-01.pkl ├── processed_v5_4.pkl ├── readme.md └── src │ ├── .DS_Store │ ├── .ipynb_checkpoints │ └── generate_target-checkpoint.ipynb │ ├── generate_prompt.ipynb │ ├── generate_target.ipynb │ ├── genomic_data_preprocess.ipynb │ ├── genomic_prompt.ipynb │ ├── processed_data │ ├── all_data.pkl │ ├── cases_gpt_trend.pkl │ ├── confimed_cases_daily_state.pkl │ ├── dynamic_prompt.pkl │ ├── final_data.pkl │ ├── genomic_prompt.pkl │ ├── hosp_gpt_trend.pkl │ ├── hospitalization_daily_state.pkl │ ├── selected_state_policy_daily.pkl │ ├── static_data.pkl │ ├── top_5_variant.pkl │ └── vaccination_weekly_state.pkl │ ├── raw_data.zip │ ├── raw_data_preprocessing.ipynb │ └── trend_descrption_with_ChatGPT.ipynb ├── env.yml └── src ├── covid_llm ├── __init__.py ├── agent.py ├── conversation.py ├── instruction_dataset.py ├── llama_flash_attn_monkey_patch.py ├── metrics.py ├── model.py └── samplers.py ├── graph_text ├── __init__.py └── prompts.py ├── scripts └── run_covid_llm_sft.py └── utils ├── __init__.py ├── basics ├── __init__.py ├── cfg_utils.py ├── iterables.py ├── logging.py ├── np_utils.py └── os_utils.py ├── data ├── __init__.py ├── covid_data.py └── prompt_tree.py ├── pkg ├── __init__.py ├── dict2xml.py ├── distributed.py └── hf_utils.py └── project ├── __init__.py └── exp.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/LICENSE -------------------------------------------------------------------------------- /PandemicLLM.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/PandemicLLM.jpg -------------------------------------------------------------------------------- /Readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/Readme.md -------------------------------------------------------------------------------- /configs/data/covid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/configs/data/covid.yaml -------------------------------------------------------------------------------- /configs/exp/sft.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/configs/exp/sft.yaml -------------------------------------------------------------------------------- /configs/llm/llama_peft.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/configs/llm/llama_peft.yaml -------------------------------------------------------------------------------- /configs/llm/llm_meta_data.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/configs/llm/llm_meta_data.yaml -------------------------------------------------------------------------------- /configs/main.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/configs/main.yaml -------------------------------------------------------------------------------- /configs/model/covid_llm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/configs/model/covid_llm.yaml -------------------------------------------------------------------------------- /configs/prompt/prompts.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/configs/prompt/prompts.yaml -------------------------------------------------------------------------------- /data/Anonymized.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/Anonymized.pkl -------------------------------------------------------------------------------- /data/data_v6_after_2021-06-01.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/data_v6_after_2021-06-01.pkl -------------------------------------------------------------------------------- /data/data_v6_after_2021-12-01.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/data_v6_after_2021-12-01.pkl -------------------------------------------------------------------------------- /data/processed_v5_4.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/processed_v5_4.pkl -------------------------------------------------------------------------------- /data/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/readme.md -------------------------------------------------------------------------------- /data/src/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/.DS_Store -------------------------------------------------------------------------------- /data/src/.ipynb_checkpoints/generate_target-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/.ipynb_checkpoints/generate_target-checkpoint.ipynb -------------------------------------------------------------------------------- /data/src/generate_prompt.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/generate_prompt.ipynb -------------------------------------------------------------------------------- /data/src/generate_target.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/generate_target.ipynb -------------------------------------------------------------------------------- /data/src/genomic_data_preprocess.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/genomic_data_preprocess.ipynb -------------------------------------------------------------------------------- /data/src/genomic_prompt.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/genomic_prompt.ipynb -------------------------------------------------------------------------------- /data/src/processed_data/all_data.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/processed_data/all_data.pkl -------------------------------------------------------------------------------- /data/src/processed_data/cases_gpt_trend.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/processed_data/cases_gpt_trend.pkl -------------------------------------------------------------------------------- /data/src/processed_data/confimed_cases_daily_state.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/processed_data/confimed_cases_daily_state.pkl -------------------------------------------------------------------------------- /data/src/processed_data/dynamic_prompt.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/processed_data/dynamic_prompt.pkl -------------------------------------------------------------------------------- /data/src/processed_data/final_data.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/processed_data/final_data.pkl -------------------------------------------------------------------------------- /data/src/processed_data/genomic_prompt.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/processed_data/genomic_prompt.pkl -------------------------------------------------------------------------------- /data/src/processed_data/hosp_gpt_trend.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/processed_data/hosp_gpt_trend.pkl -------------------------------------------------------------------------------- /data/src/processed_data/hospitalization_daily_state.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/processed_data/hospitalization_daily_state.pkl -------------------------------------------------------------------------------- /data/src/processed_data/selected_state_policy_daily.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/processed_data/selected_state_policy_daily.pkl -------------------------------------------------------------------------------- /data/src/processed_data/static_data.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/processed_data/static_data.pkl -------------------------------------------------------------------------------- /data/src/processed_data/top_5_variant.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/processed_data/top_5_variant.pkl -------------------------------------------------------------------------------- /data/src/processed_data/vaccination_weekly_state.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/processed_data/vaccination_weekly_state.pkl -------------------------------------------------------------------------------- /data/src/raw_data.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/raw_data.zip -------------------------------------------------------------------------------- /data/src/raw_data_preprocessing.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/raw_data_preprocessing.ipynb -------------------------------------------------------------------------------- /data/src/trend_descrption_with_ChatGPT.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/data/src/trend_descrption_with_ChatGPT.ipynb -------------------------------------------------------------------------------- /env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/env.yml -------------------------------------------------------------------------------- /src/covid_llm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/covid_llm/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/covid_llm/agent.py -------------------------------------------------------------------------------- /src/covid_llm/conversation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/covid_llm/conversation.py -------------------------------------------------------------------------------- /src/covid_llm/instruction_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/covid_llm/instruction_dataset.py -------------------------------------------------------------------------------- /src/covid_llm/llama_flash_attn_monkey_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/covid_llm/llama_flash_attn_monkey_patch.py -------------------------------------------------------------------------------- /src/covid_llm/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/covid_llm/metrics.py -------------------------------------------------------------------------------- /src/covid_llm/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/covid_llm/model.py -------------------------------------------------------------------------------- /src/covid_llm/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/covid_llm/samplers.py -------------------------------------------------------------------------------- /src/graph_text/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/graph_text/prompts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/graph_text/prompts.py -------------------------------------------------------------------------------- /src/scripts/run_covid_llm_sft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/scripts/run_covid_llm_sft.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/utils/__init__.py -------------------------------------------------------------------------------- /src/utils/basics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/utils/basics/__init__.py -------------------------------------------------------------------------------- /src/utils/basics/cfg_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/utils/basics/cfg_utils.py -------------------------------------------------------------------------------- /src/utils/basics/iterables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/utils/basics/iterables.py -------------------------------------------------------------------------------- /src/utils/basics/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/utils/basics/logging.py -------------------------------------------------------------------------------- /src/utils/basics/np_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/utils/basics/np_utils.py -------------------------------------------------------------------------------- /src/utils/basics/os_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/utils/basics/os_utils.py -------------------------------------------------------------------------------- /src/utils/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/data/covid_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/utils/data/covid_data.py -------------------------------------------------------------------------------- /src/utils/data/prompt_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/utils/data/prompt_tree.py -------------------------------------------------------------------------------- /src/utils/pkg/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/pkg/dict2xml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/utils/pkg/dict2xml.py -------------------------------------------------------------------------------- /src/utils/pkg/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/utils/pkg/distributed.py -------------------------------------------------------------------------------- /src/utils/pkg/hf_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/utils/pkg/hf_utils.py -------------------------------------------------------------------------------- /src/utils/project/__init__.py: -------------------------------------------------------------------------------- 1 | from .exp import * 2 | -------------------------------------------------------------------------------- /src/utils/project/exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/miemieyanga/PandemicLLM/HEAD/src/utils/project/exp.py --------------------------------------------------------------------------------