我有一个数据集,其中包含大量类别为0的实例和极少数类别为1的实例,这是一个问题,因为类别为0的实例会主导类别为1的实例,导致类别1的精度非常低。我正在使用Weka的Java API,在API中我发现了一个给实例赋予权重的选项,因此我决定在测试集中对类别为1的实例赋予权重,具体操作如下:
breader = new BufferedReader(new FileReader("weka/train.txt")); Instances train = new Instances(breader); train.setClassIndex(train.numAttributes() - 1); Instances testset= new Instances(new BufferedReader(new FileReader("weka/test.txt"))); unlabeled.setClassIndex(testset.numAttributes() - 1); for (int i = 0; i < testset.numInstances(); i++) { if(testset.instance(i).classValue()==1){ testset.instance(i).setWeight(30); } }
这样做之后,精度大幅提高。现在我想知道我这样做是否可以接受,如果可以,如何解释这种做法?
回答:
你必须考虑到,给实例添加的权重会影响你的预测模型。对于这些情况,预测模型也会变得很重。由于这种不寻常的训练,你可能会陷入过拟合。过拟合的可能性存在,因为用于训练模型的标准可能与用于评估模型效能的标准不同。然而,如果你无法获得更多的训练数据,这是你可以承担的风险。毕竟,这对你有效。