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.