Python Cross Entropy Loss Machine Learning