这个模型是在 UA-SQuAD 数据集上对 ukr-models/xlm-roberta-base-uk 进行微调得到的。
训练脚本链接 - https://github.com/robinhad/ukrainian-qa 它在评估集上取得以下结果:
需要更多信息
from transformers import pipeline, AutoTokenizer, AutoModelForQuestionAnswering model_name = "robinhad/ukrainian-qa" tokenizer = AutoTokenizer.from_pretrained(model_name) model = AutoModelForQuestionAnswering.from_pretrained(model_name) qa_model = pipeline("question-answering", model=model.to("cpu"), tokenizer=tokenizer) question = "Де ти живеш?" context = "Мене звати Сара і я живу у Лондоні" qa_model(question = question, context = context)
需要更多信息
需要更多信息
训练过程中使用了以下超参数:
Training Loss | Epoch | Step | Validation Loss |
---|---|---|---|
2.4526 | 1.0 | 650 | 1.3631 |
1.3317 | 2.0 | 1300 | 1.2229 |
1.0693 | 3.0 | 1950 | 1.2184 |
0.6851 | 4.0 | 2600 | 1.3171 |
0.5594 | 5.0 | 3250 | 1.3893 |
0.4954 | 6.0 | 3900 | 1.4778 |