def __init__(self, input):
from pymc.Model import Model
from pymc.Node import ZeroProbability
self.model = Model(input)
# sort parameters by name
ordered_pairs = sorted((s.__name__, s) for s in self.model.stochastics)
# Record parameter, shape, size, offset as a list
pars = [v for k, v in ordered_pairs]
shape = [v.shape for v in pars]
size = [(np.prod(s) if s != () else 1) for s in shape]
offset = np.cumsum([0]+size[:-1])
self.pars = list(zip(pars, shape, size, offset))
# List of cost functions contains both parameter and data, but not
# intermediate deterministic points
self.costs = self.model.variables - self.model.deterministics
# Degrees of freedom is #data - #pars
points = sum((np.prod(p.shape) if p.shape != () else 1)
for p in self.costs)
self.dof = points - 2*offset[-1]
self.ZeroProbability = ZeroProbability
class PyMCProblem(object):
def __init__(self, input):
from pymc.Model import Model
from pymc.Node import ZeroProbability
self.model = Model(input)
# sort parameters by name
ordered_pairs = sorted((s.__name__, s) for s in self.model.stochastics)
# Record parameter, shape, size, offset as a list
pars = [v for k, v in ordered_pairs]
shape = [v.shape for v in pars]
size = [(np.prod(s) if s != () else 1) for s in shape]
offset = np.cumsum([0]+size[:-1])
self.pars = list(zip(pars, shape, size, offset))
# List of cost functions contains both parameter and data, but not
# intermediate deterministic points
self.costs = self.model.variables - self.model.deterministics
# Degrees of freedom is #data - #pars
points = sum((np.prod(p.shape) if p.shape != () else 1)
for p in self.costs)
self.dof = points - 2*offset[-1]
self.ZeroProbability = ZeroProbability
def model_reset(self):
pass
def chisq(self):
return self.nllf() # /self.dof
def chisq_str(self):
return "%g"%self.chisq()
__call__ = chisq
def nllf(self, pvec=None):
if pvec is not None: self.setp(pvec)
try:
return -sum(c.logp for c in self.costs)
except self.ZeroProbability:
return inf
def setp(self, values):
for par, shape, size, offset in self.pars:
if shape == ():
par.value = values[offset]
offset += 1
else:
par.value = array(values[offset:offset+size]).reshape(shape)
offset += size
def getp(self):
return np.hstack([(par.value.flatten() if shape != () else par.value)
for par, shape, size, offset in self.pars])
def show(self):
# maybe print graph of model
print("[chisq=%g, nllf=%g]" % (self.chisq(), self.nllf()))
print(self.summarize())
def summarize(self):
return "\n".join("%s=%s"%(par.__name__, par.value)
for par, _, _, _ in self.pars)
def labels(self):
ret = []
for par, _, _, _ in self.pars:
ret.extend(_par_labels(par))
return ret
def randomize(self, N=None):
if N is None:
self.model.draw_from_prior()
else:
data = []
for _ in range(N):
self.model.draw_from_prior()
data.append(self.getp())
return asarray(data)
def bounds(self):
return np.vstack([_par_bounds(par) for par, _, _,_ in self.pars]).T
def plot(self, p=None, fignum=None, figfile=None):
pass
def __deepcopy__(self, memo):
return self
