#!/usr/bin/env python

from pylab import *
from dream import *

x = linspace(-4.0, 4, 40)
fn = lambda p: polyval(p, x)
bounds = (-20, -20, -20), (40, 40, 40)
sigma = 1
data = fn((2, -1, 3)) + randn(*x.shape) * sigma  # Fake data


n = len(bounds[0])
model = Simulation(f=fn, data=data, sigma=sigma, bounds=bounds)
sampler = Dream(
    model=model,
    population=randn(5 * n, 4, n),
    thinning=1,
    draws=20000,
)

mc = sampler.sample()
figure(6)
model.plot(mc.best()[0])
plot_all(mc)