Views
No views yet
1README.md
2predict_ensemble.py
3dataset/
4 SemEval/
5bert-base-uncased/ # 从 Hugging Face 下载
6models/ # 从 Hugging Face 下载
7 rbert_lr2e-5_len192_seed1.pt
8 rbert_lr2e-5_len192_seed2.pt
9 rbert_lr2e-5_len192_seed3.pt
10 rbert_lr2e-5_len192_seed4.pt
11 rbert_lr2e-5_len192_seed5.pt
12loss_curves/
13 rbert_lr2e-5_len192_seed1_loss.png
14 rbert_lr2e-5_len192_seed2_loss.png
15 rbert_lr2e-5_len192_seed3_loss.png
16 rbert_lr2e-5_len192_seed4_loss.png
17 rbert_lr2e-5_len192_seed5_loss.pngmodels/ 中是训练后的 5 个模型权重。predict_ensemble.py 是测试集预测代码,默认加载这 5 个模型,在测试集上分别预测,然后对每个样本做多数投票,输出最终预测标签。https://huggingface.co/liuyanliang/rbert-relation-extraction-semevalbert-base-uncased/ 也已上传到同一个 Hugging Face 仓库内。dataset/SemEval/ 已经放在提交目录内。运行预测前需要先下载 models/ 权重和 bert-base-uncased/。pip install torch==2.12.0 numpy==2.4.4 transformers==5.9.01cd submission_L_relation_extraction
2
3hf download liuyanliang/rbert-relation-extraction-semeval \
4 --include "models/*.pt" \
5 --include "bert-base-uncased/*" \
6 --local-dir .1python predict_ensemble.py \
2 --bert-model ./bert-base-uncased \
3 --data-root ./dataset/SemEval/ \
4 --vector-path ./dataset/SemEval/vector_50.txt \
5 --model-dir ./models \
6 --max-len 192 \
7 --batch-size 16 \
8 --pred-path pred_rbert_seed_ensemble_vote.txtpred_rbert_seed_ensemble_vote.txt1bert_model = ./bert-base-uncased
2data_root = ./dataset/SemEval/
3epochs = 20
4batch_size = 16
5learning_rate = 2e-5
6dropout = 0.2
7patience = 8
8max_len = 192
9weight_decay = 0.01
10seed = 1, 2, 3, 4, 5dev_macroF1 作为保存最佳 checkpoint 和 early stopping 的指标。



