Tensorflow Model Compile Loss