"Invalid classification data: expect label value"

Question

我正在尝试在 java 中使用深度学习训练模型，当我开始训练训练数据时出现错误

Invalid classification data: expect label value (at label index column = 0) to be in range 0 to 1 inclusive (0 to numClasses-1, with numClasses=2); got label value of 2

由于我是深度学习4j的初学者，所以我没有理解错误。我正在使用一个查看两个人之间关系的数据集（如果两个人之间存在关系，那么 class 标签将为 1，否则为 0）。

Java代码

public class SNA {
private static Logger log = LoggerFactory.getLogger(SNA.class);

public static void main(String[] args) throws Exception {
    int seed = 123;
    double learningRate = 0.01;
    int batchSize = 50;
    int nEpochs = 30;
    int numInputs = 2;
    int numOutputs = 2;
    int numHiddenNodes = 20;

    //load the training data
    RecordReader rr = new CSVRecordReader(0,",");
    rr.initialize(new FileSplit(new File("C:\Users\GTS\Desktop\SNA project\experiments\First experiment\train\slashdotTrain.csv")));
    DataSetIterator trainIter = new RecordReaderDataSetIterator(rr, batchSize,0, 2);

    // load test data
    RecordReader rrTest = new CSVRecordReader();
    rr.initialize(new FileSplit(new File("C:\Users\GTS\Desktop\SNA project\experiments\First experiment\test\slashdotTest.csv")));
    DataSetIterator testIter = new RecordReaderDataSetIterator(rrTest, batchSize,0, 2);

    log.info("**** Building Model ****");
    MultiLayerConfiguration conf = new NeuralNetConfiguration.Builder()
            .seed(seed)
            .optimizationAlgo(OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT)
            .iterations(1)
            .learningRate(learningRate)
            .updater(Updater.NESTEROVS).momentum(0.9)
            .list()
            .layer(0, new DenseLayer.Builder()
                    .nIn(numInputs)
                    .nOut(numHiddenNodes)
                    .activation("relu")
                    .weightInit(WeightInit.XAVIER)
                    .build())
            .layer(1, new OutputLayer.Builder(LossFunctions.LossFunction.NEGATIVELOGLIKELIHOOD)
                    .activation("softmax")
                    .weightInit(WeightInit.XAVIER)
                    .nIn(numHiddenNodes)
                    .nOut(numOutputs)
                    .build())
            .pretrain(false).backprop(true)
            .build();

    MultiLayerNetwork model = new MultiLayerNetwork(conf);
    model.init();

    // Listener to show how the network is training in the log
    model.setListeners(new ScoreIterationListener(10));

    log.info(" **** Train Model **** ");
    for (int i = 0; i < nEpochs; i++) {
        model.fit(trainIter);
    }

    System.out.println("**** Evaluate Model ****");
    Evaluation evaluation = new Evaluation(numOutputs);
    while (testIter.hasNext()) {
        DataSet t = testIter.next();
        INDArray feature = t.getFeatureMatrix();
        INDArray labels = t.getLabels();
        INDArray predicted = model.output(feature, false);
        evaluation.eval(labels, predicted);
    }

    System.out.println(evaluation.stats());
}

}

有什么帮助吗？非常感谢

Answer 1

问题已解决：把RecordReaderDataSetIterator的第三个参数改成
DataSetIterator testIter = new RecordReaderDataSetIterator(rrTest, batchSize,0, 2); 从 0 到 2；因为数据集有三列并且 class 标签的索引是 2 因为它是第三列。

解决方案：

DataSetIterator trainIter = new RecordReaderDataSetIterator(rr, batchSize,2, 2);

参考文献： enter link description here

"Invalid classification data: expect label value"

"Invalid classification data: expect label value"

java

neural-network

deeplearning4j