Hongyang449/covid19_perception

Text-based prediction of COVID-19

NLP models are used in predictiong COVID-19 based on texts about pereceptual ability changes in smell, taste, and chemesthesis.

---

Installation

Git clone a copy of code:

git clone https://github.com/Hongyang449/covid19_perception.git

Required dependencies

• python (3.9.7)
• transformers (4.21.0)
• shap (0.39.0)

1. Preprocess data

• directory data/
• python preprocess.py

2. Train NLP models and make predictions

• directories code/class5/ and code/class6/
• python train.py

3. SHAP analysis

• directories code/class5/ and code/class6/
• python shap_analysis.py
• python shap_consensus.py