He.ColdThinking 2021-11-24 17:02 采纳率: 0%
浏览 27
已结题

transformers库中的Trainer API中没有设计得到train acc吗？请教怎么得到train acc

使用huggingface 的transformers库进行预训练模型的训练时，发现只能得到train loss。
而且compute_metrics好像只能计算验证的acc和loss。

  • 写回答

1条回答 默认 最新

  • 爱晚乏客游 2021-11-25 10:00
    关注

    如果你需要train acc，那么在Trainer中将eval_dataset指定成train_data,也就是让验证集变成训练集就是了，然后compute_metrics一样，就算的就是训练集的了出来的就是train acc和train loss，但是这样没啥意义

    评论

报告相同问题？

问题事件

  • 系统已结题 12月2日
  • 创建了问题 11月24日