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