I have a ML models which have 3 output. Each of them have passed into a SoftMax player. I\'m trying to train the model from custom scratch with training loop in keras. My genera