Example usage for org.deeplearning4j.nn.multilayer MultiLayerNetwork getOutputLayer

Introduction

In this page you can find the example usage for org.deeplearning4j.nn.multilayer MultiLayerNetwork getOutputLayer.

Prototype

public Layer getOutputLayer() 

Source Link

Document

Get the output layer - i.e., the last layer in the netwok

Usage

From source file:org.audiveris.omr.classifier.DeepClassifier.java

License:Open Source License

@Override
protected boolean isCompatible(MultiLayerNetwork model, Norms norms) {
    // Check input numbers for norms
    final int normsIn = norms.means.columns();

    if (normsIn != 1) {
        logger.warn("Incompatible norms count:{} expected:{}", normsIn, 1);

        return false;
    }//from   w  w w.  j  av  a2 s  . c o m

    // Check input numbers for model
    final org.deeplearning4j.nn.layers.convolution.ConvolutionLayer inputLayer = (org.deeplearning4j.nn.layers.convolution.ConvolutionLayer) model
            .getLayer(0);
    final org.deeplearning4j.nn.conf.layers.ConvolutionLayer confInputLayer = (org.deeplearning4j.nn.conf.layers.ConvolutionLayer) inputLayer
            .conf().getLayer();
    final int modelIn = confInputLayer.getNIn();

    if (modelIn != 1) {
        logger.warn("Incompatible features count:{} expected:{}", modelIn, 1);

        return false;
    }

    // Check output numbers for model
    final org.deeplearning4j.nn.layers.OutputLayer outputLayer = (org.deeplearning4j.nn.layers.OutputLayer) model
            .getOutputLayer();
    final org.deeplearning4j.nn.conf.layers.OutputLayer confOutputLayer = (org.deeplearning4j.nn.conf.layers.OutputLayer) outputLayer
            .conf().getLayer();
    final int modelOut = confOutputLayer.getNOut();

    if (modelOut != SHAPE_COUNT) {
        logger.warn("Incompatible shape count model:{} expected:{}", modelOut, SHAPE_COUNT);

        return false;
    }

    return true;
}

---