Не могу передать слушателю коллекцию, DL4J

Question

Я впервые работаю с dl4j, так что будьте осторожны со мной.

Я написал следующую простую программу

import org.datavec.api.records.reader.RecordReader;
import org.datavec.api.records.reader.impl.csv.CSVRecordReader;
import org.datavec.api.split.FileSplit;
import org.deeplearning4j.datasets.datavec.RecordReaderDataSetIterator;
import org.deeplearning4j.nn.api.OptimizationAlgorithm;
import org.deeplearning4j.nn.conf.MultiLayerConfiguration;
import org.deeplearning4j.nn.conf.NeuralNetConfiguration;
import org.deeplearning4j.nn.conf.Updater;
import org.deeplearning4j.nn.conf.layers.DenseLayer;
import org.deeplearning4j.nn.conf.layers.OutputLayer;
import org.deeplearning4j.nn.multilayer.MultiLayerNetwork;
import org.deeplearning4j.nn.weights.WeightInit;
import org.deeplearning4j.optimize.listeners.ScoreIterationListener;
import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.dataset.DataSet;
import org.nd4j.linalg.dataset.api.iterator.DataSetIterator;
import org.nd4j.linalg.lossfunctions.LossFunctions;
import org.deeplearning4j.eval.Evaluation;

import java.io.File;
import java.util.Collection;


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

        //Load Training Data
        RecordReader rr = new CSVRecordReader();
        rr.initialize(new FileSplit(new File("C:\\Users\\Oria\\MLP\\linear_data_train.csv")));
        DataSetIterator trainIter = new RecordReaderDataSetIterator(rr, batchSize,0,2);

        //Load Testing Data
        RecordReader rrTest = new CSVRecordReader();
        rrTest.initialize(new FileSplit(new File("C:\\Users\\Oria\\MLP\\linear_data_eval.csv")));
        DataSetIterator testIter = new RecordReaderDataSetIterator(rrTest, batchSize,0,2);

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

        MultiLayerNetwork model = new MultiLayerNetwork(conf);
        model.init();
        model.setListeners(new ScoreIterationListener(10));

        for(int i = 0; i < nEpochs; i++)
            model.fit(trainIter);

        System.out.println("Evaluate model.......");
        Evaluation eval = new Evaluation(numOutputs);
        while(testIter.hasNext())
        {
            DataSet t = testIter.next();
            INDArray features = t.getFeatureMatrix();
            INDArray lables = t.getLabels();
            INDArray predicted = model.output(features,false);
            eval.eval(lables,predicted);
        }
        System.out.println(eval.stats());
    }
}

Код должен работать нормально. Это копия https://www.youtube.com/watch?v=8EIBIfVlgmU&t=1063s которая является хорошо известным руководством по dl4j.

Тем не менее, код не компилируется. Я получаю сообщение об ошибке в строке model.SetListeners: "Метод setListeners(Collection) в типе MultiLayerNetwork не применим для аргументов (ScoreIterationListener)"

Когда я изменяю его на "model.setListeners((Collection) new ScoreIteredListener(10));" ошибка компиляции исчезает, но вместо этого я получаю ошибку времени выполнения "Исключение в потоке"main" java.lang.ClassCastException: org.deeplearning4j.optimize.listeners.ScoreIterationListener нельзя привести к java.util.Collection в MLPClassifierLinear.main(MLPClassifierLinearar.java:71)"

В чем дело? Кто-нибудь, имеющий опыт работы с dl4j, может помочь мне решить эту проблему?

Answer 1

Я хотел бы начать с наших примеров и убедиться, что вы используете последнюю версию (0.7.2 на момент написания этой статьи): https://github.com/deeplearning4j/dl4j-examples

MultiLayerNetwork.setListeners имеет переменные аргументы в IterationListener: https://github.com/deeplearning4j/deeplearning4j/blob/master/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/multilayer/MultiLayerNetwork.java#L1232

У меня такое чувство, что настоящей причиной может быть что-то еще?