Я строю модель логистической регрессии, но получаю ошибку java.lang.ArrayIndexOutOfBoundsException

Question

Я строю lr, используя deeplearning4j, и только три тренировочных данных.

Но я получаю ошибку:

Exception in thread "main" java.lang.ArrayIndexOutOfBoundsException: 1

double [][] train = new double[][]{{1.2,2.2,3.2},{2.2,3.2,4.2},{4.5,5.2,4.1}};
int [] label = {1,1,0};
        INDArray trainSet = Nd4j.create(train);
        net.fit(trainSet,label);
And the layer is:
OutputLayer outputLayer = new OutputLayer.Builder(LossFunction.XENT)
                .nIn(dim)
                .nOut(1)
                .weightInit(WeightInit.XAVIER)
                .activation(Activation.SIGMOID)
                .build();
 MultiLayerConfiguration logistic = new NeuralNetConfiguration.Builder()
                .seed(123)
                .optimizationAlgo(OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT)
                .list() 
                .layer(0,outputLayer) 
                .build(); 

что-то не так с моимкод слоя? И я хочу сделать бинарную классификацию, правильно ли, что метка 0 и 1