Java Deep Learning Cookbook

Constructing output layers for output classification

We need to perform image classification using logistic regression (SOFTMAX), resulting in probabilities of occurrence for each of the image labels. Logistic regression is a predictive analysis algorithm and, hence, more suitable for prediction problems. In this recipe, we will design output layers for the image classification problem.