simpleplanets_kepler_known_test.py

from simpleabc import simple_abc
import simple_model
import numpy as np
import pickle
from scipy import stats
import time


steps = 10
eps = 1
min_part = 100

stars = pickle.load(file('stars.pkl'))

model = simple_model.MyModel(stars)

#obs = np.recfromcsv('04012015_trimmed.csv',usecols=(1,4,14,32),delimiter=",")
#obs = obs[obs['koi_disposition'] != "FALSE POSITIVE"]
#obs.dtype.names = 'ktc_kepler_id','koi_disposition','period', 'T'

theta_0 = (2.0, 0.1, 9)
obs = model.generate_data(theta_0)

model.set_prior([stats.uniform(0, 90.0),
                 stats.uniform(0, 1),
                 stats.uniform(0, 20)])

model.set_data(obs)

start = time.time()
OT = simple_abc.pmc_abc(model, obs, epsilon_0=eps, min_particles=min_part,
                        steps=2, target_epsilon=eps, parallel=False)
out_pickle = file('pickles/kepler_pmc_xi_based_each_bin_dist_100_KNOWN_prime.pkl', 'w')
pickle.dump(OT, out_pickle)
out_pickle.close()

for i in range(0, steps):
    PT = OT
    OT = simple_abc.pmc_abc(model, obs, epsilon_0=eps, min_particles=min_part,
                        resume=PT, steps=2, target_epsilon=eps, parallel=False)
    out_pickle = file('pickles/kepler_pmc_xi_based_each_bin_dist_100_KNOW__{:}.pkl'.format(i),
                      'w')
    pickle.dump(OT, out_pickle)
    out_pickle.close()

end = time.time()
print 'Serial took {}s'.format(end - start)