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