Как мне хранить и загружать обученные данные в deeplearning4j? - PullRequest
0 голосов
/ 20 декабря 2018

deeplearning4j: Как я могу сохранить / сохранить обученную модель на уровне постоянства и загрузить ее обратно, когда приходит специальный запрос для оценки модели глубокого обучения?

        DataNormalization normalizer = new NormalizerStandardize();
        normalizer.fit(trainingData);           //Collect the statistics (mean/stdev) from the training data. This does not modify the input data
        normalizer.transform(trainingData); 

        //run the model
        MultiLayerNetwork model = new MultiLayerNetwork(conf);
        model.init();
        model.setListeners(new ScoreIterationListener(100));

        for( int i=0; i<epochs; i++ ) {
            model.fit(trainingData);
        }

Мне нужнохранить обученную модель.Как я могу это сделать?С каким Api?

        //evaluate the model on the test set
        Evaluation eval = new Evaluation(3);
        INDArray output = model.output(testData.getFeatures());

        eval.eval(testData.getLabels(), output);
        log.info(eval.stats());    

1 Ответ

0 голосов
/ 04 января 2019

С ModelSerializer

Вы можете написать / прочитать это так

ModelSerializer.writeModel(modelToSave, "location", true);

...

MultiLayerNetwork model = ModelSerializer.restoreMultiLayerNetwork("location");
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...