EasyNLP icon indicating copy to clipboard operation
EasyNLP copied to clipboard

使用K-bert做文本分类无法指定分类类别

Open shishijier opened this issue 1 year ago • 1 comments

通过修改 examples/kbert/run_cls.sh 这个文件下的 label_enumerate_values 改变模型分类个数。 自己数据集一共是10种类别,值为0-9,但是运行会报错。 通过打印logits 的值发现,模型输出依然是二分类。logits形状为[batch_size,2] 如何改变分类类别?

shishijier avatar Jun 07 '23 13:06 shishijier