This project is based on the backbone repository: https://github.com/TobiasLee/CCPM-baseline.
pip install -r requirements.txt
- Modifying the module importing part of main.py Line 52 (Optional)
from model_allcat_tag import unwrapped_preprocess_function, MyModule, DataCollatorForMultipleChoice, MyTokenizer, MyOptimizer
-
Modifying the parameters and available CUDA device in run_model.sh (Optional)
chmod +x run_model.sh
./run_model.sh
zip
├── CCPM-data ### Original and split data
│ ├── split_test.jsonl
│ ├── split_valid.jsonl
│ ├── test_public.jsonl
│ ├── train.jsonl
│ └── valid.jsonl
├── data_split.py ### code for splitting valid.jsonl
├── main.py ### entry of all models except PLM-Match
├── main_sim.py ### entry of PLM-Match model
├── model_allcat.py ### PLM-All
├── model_allcat_tag.py ### PLM-All-Tag
├── model.py ### PLM-CLS
├── model_sim.py ### PLM-Match
├── README.md
├── requirements.txt
├── run_model.sh