def testNegBinomWSd(k, r, mu, sd): truth = mean([nbinom.logpmf(k, i, mu) for i in range(int(r-0.5*sd),int(r+0.5*sd)+1)]) val = u.log_neg_binom_likelihood(k,r,mu,sd) print truth, val return truth==val
def testNegBinomNoSd(k, r, mu): truth = nbinom.logpmf(k,r,mu) val = u.log_neg_binom_likelihood(k,r,mu) return truth==val
def calcLogLikelihoodNegBinom(self, mu, tp, sd_d=0): return u.log_neg_binom_likelihood(self.a[tp], self.d[tp]-self.a[tp], mu, sd_d)