calculate_loss function

Calculate Loss Function