Added trained parameters for softmax