Save & Load

public Model saveModel() throws IOException {
        log.info("Save model............");
        OutputStream fos = Files.newOutputStream(Paths.get(Config.FILE_PARAMS));
        DataOutputStream dos = new DataOutputStream(fos);
        Nd4j.write(model.params(), dos);
        dos.flush();
        dos.close();
        FileUtils.writeStringToFile(new File(Config.FILE_MODEL), model.getLayerWiseConfigurations().toJson());
        return this;
    }
    
    private MultiLayerNetwork loadModel() {
        log.info("Load model...........");

        try {
            MultiLayerConfiguration confFromJson = MultiLayerConfiguration.fromJson(FileUtils.readFileToString(new File(FileHelper.getFilePath(Config.FILE_MODEL))));
            DataInputStream dis = new DataInputStream(new FileInputStream(FileHelper.getFilePath(Config.FILE_PARAMS)));
            INDArray newParams = Nd4j.read(dis);
            dis.close();
            MultiLayerNetwork savedNetwork = new MultiLayerNetwork(confFromJson);
            savedNetwork.init();
            savedNetwork.setParameters(newParams);
            System.out.println("Num params: "+savedNetwork.numParams());
            return savedNetwork;
        } catch (Exception ex) {
            ex.printStackTrace();
        }
        return null;
    }