SimpleTransformers默认配置运行失败 报ValueError文本需为str类型求助

问题原因 #

你的训练数据中target_text列的二分类标签为整数类型（1、0），并非字符串类型。T5是生成式模型，要求所有输入、输出字段均为字符串格式，新版本的transformers库增加了严格的输入类型校验，传入非字符串类型的文本会直接触发该报错。你之前检查字段类型时大概率忽略了分类任务的标签值类型。

修复方案 #

在构建好训练、验证DataFrame后，统一将target_text列转换为字符串类型即可，修改代码如下：

# 新增在train_df定义完成之后
train_df['target_text'] = train_df['target_text'].astype(str)
# 后续要用的eval_df也同步转换
eval_df['target_text'] = eval_df['target_text'].astype(str)

验证方法 #

执行print(train_df.dtypes)查看各列类型，若target_text列显示为object则表示转换成功，可正常运行训练代码。

内容的提问来源于stack exchange，提问作者brettfazio