评价指标问题。能否将训练阶段和预测阶段的评价指标修改为accuracy和F1之类的。 #6596
-
|
我想将大模型微调成一个多分类任务的分类器，通过Input和指令，直接进行多分类，输出0,1这样的数字。微调阶段训练的数据的output也都是0,1数字，有没有什么方法在训练或者预测阶段让评价结果修改为accuracy和F1呢？ |
Beta Was this translation helpful? Give feedback.
Answered by
hiyouga
Jan 10, 2025
Replies: 1 comment
-
|
自行修改 https://github.com/hiyouga/LLaMA-Factory/blob/main/src/llamafactory/train/sft/metric.py |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
KyleLeith-007
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
自行修改 https://github.com/hiyouga/LLaMA-Factory/blob/main/src/llamafactory/train/sft/metric.py