Negative Log Likelihood Python