I trained a Mnist model with DL4J.
When i use this model in inference mode :
INDArray prediction = myModel.output(myINDArrayImage);
that gives me a prediction in an INDArray, it works properly. The size of this INDArray is equal to number of output on my OutputLayer model (softmax activation).
Is there a possibility to ask the model to predict only on a certain number ? Or have I to change my model ?
thx for your responses !