8igfive / HyperSent

A project for Information Retrieval course.

环境

使用的是 conda 环境，所用到的包参考 requirements.txt

数据

训练数据位于 data/ 中，可以通过以下指令下载：

bash data/download_nli.sh

测试数据位于 SentEval/data/downstream 中，可以通过以下指令下载：

bash SentEval/data/downstream/download_dataset.sh

模型训练

可以通过以下脚本运行训练示例：

bash scripts/training.sh

测试结果

python evaluation.py \
  --model_name_or_path results/runs/test_training/checkpoint-4750 \
  --model_type bert