prediction returns Nx1 vectors

This commit is contained in:
Ricardo 2013-05-15 18:13:49 +01:00
parent df9700c412
commit 639b6a6b9d

View file

@ -58,7 +58,7 @@ class probit(likelihood_function):
norm_975 = [stats.norm.ppf(.975,m,v) for m,v in zip(mu,var)] norm_975 = [stats.norm.ppf(.975,m,v) for m,v in zip(mu,var)]
p_025 = stats.norm.cdf(norm_025/np.sqrt(1+var)) p_025 = stats.norm.cdf(norm_025/np.sqrt(1+var))
p_975 = stats.norm.cdf(norm_975/np.sqrt(1+var)) p_975 = stats.norm.cdf(norm_975/np.sqrt(1+var))
return mean, np.nan*var, p_025, p_975 # TODO: var return mean[:,None], np.nan*var, p_025[:,None], p_975[:,None] # TODO: var
class Poisson(likelihood_function): class Poisson(likelihood_function):
""" """