FIX: Add softmax to MLP multi-class output
FIX: Add softmax to MLP multi-class output

Previously the softmax function was only applied MLP multi-class
output when evaluating, not during training.

gganis/root,gganis/root,gganis/root,gganis/root,gganis/root,gganis/root,gganis/root,gganis/root,gganis/root,gganis/root,gganis/root