Extracting Target Labels from Deep Learning Classification Models

Extracting Target Labels from Deep Learning Classification Models - Enthought, Inc.

In the blog post Configuring a Neural Network Output Layer I highlighted how to correctly set up an output layer for deep learning models. Here, I discuss how to make sense of what a neural network actually returns from the output layers. If you are like me, you may have been surprised when you first encountered the output of a simple classification neural net.

“Wait… I thought this was supposed to give me an array of integers that was similar to my target variable. Why am I seeing floating point numbers? What do these numbers even mean?”

In this post, I’ll answer these types of questions and more by showing how to extract target labels from deep learning classification models.