search

zjunlp / IEPile

[ACL 2024] IEPile: A Large-Scale Information Extraction Corpus
http://oneke.openkg.cn/
Other
169 stars 16 forks source link

KeyError: 'response' #13

Closed wuQi-666 closed 6 months ago

wuQi-666 commented 6 months ago

利用https://huggingface.co/datasets/zjunlp/InstructIE/tree/main/InstructIE-zh/InstructIE_%E5%8C%BB%E5%AD%A6 中的数据集在qwen1.5-14b-chat的lora微调模型的基础上继续训练，其中的运行代码.bash中的参数设置如下所示 Snipaste_2024-05-06_17-06-01 运行报错如下所示 image 其中使用的数据集已经经过格式的转换

guihonghao commented 6 months ago

如果要将test集作为训练时的验证集，那么在处理test集时也应该使用 --split train，如下的命令转换数据格式。

python ie2instruction/convert_func.py \
    --src_path data/NER/sample.json \
    --tgt_path data/NER/train.json \
    --schema_path data/NER/schema.json \
    --language zh \
    --task NER \
    --split_num 6 \       
    --random_sort \
    --split train
wuQi-666 commented 6 months ago

谢谢，问题已经解决，代码可以继续训练