change to double
normalized_loss
Output prob for training
Add the train tool