LLM outputs.loss 返回什么
以文本分类任务(二分类,如情感分析,判断影评是正面还是负面 )为例,来解释 outputs.loss
及其计算方式。

1. 任务设定与模型准备
假设有一个影评数据集,其中包含影评文本及其对应的情感标签(0表示负面,1表示正面 )。使用 transformers
库中的 AutoModelForSequenceClassification
模型,这里以 bert-base-uncased
预训练模型为基础。
2. 模型输出结构
当调用模型进行计算时,如:
outputs = model(input_ids