Machine Learning, Transformer, Multi-class class化,测试数据和训练数据classes个数不一致

Machine Learning, Transformer, Multi-class classification, number of classes is inconsistent in test data and training data

例如,假设我正在构建一个变换器模型(来自 huggingface)并且训练数据中有 20 类,但测试数据中只有 5 类。要从 huggingface 配置 transformer 模型,例如 BertConfig,我们需要提供一个参数:num_labels。我应该将 num_labels 设置为 20 还是 5,因为测试数据中只有 5 类?

20.because 你训练的网络有 20 类.