Icon

Map Softmax Output to Class Names Example

This workflow shows how to map softmax output of a classification network to class names.

Create mapping of class index to class name (they are the same in this case because we predict numbers) MNIST imagesMNIST imagessimple, untrainedCNNinput: untrained net andtraining images with labelsoutput: trained net- -5 epochs(increase to improve accuracy)Find prediction with highest probability Preparetraining data Prepare test data DL PythonNetwork Creator Keras NetworkLearner Keras NetworkExecutor Column Expressions ExtractColumn Header Transpose Column Filter Counter Generation Column Rename Joiner RowID Create mapping of class index to class name (they are the same in this case because we predict numbers) MNIST imagesMNIST imagessimple, untrainedCNNinput: untrained net andtraining images with labelsoutput: trained net- -5 epochs(increase to improve accuracy)Find prediction with highest probabilityPreparetraining data Prepare test data DL PythonNetwork Creator Keras NetworkLearner Keras NetworkExecutor Column Expressions ExtractColumn Header Transpose Column Filter Counter Generation Column Rename Joiner RowID

Nodes

Extensions

Links