import sys sys.path.append("..") from barry.config import setup from barry.models import PowerBeutler2017 from barry.datasets import PowerSpectrum_SDSS_DR12_Z061_NGC from barry.samplers import DynestySampler from barry.fitter import Fitter if __name__ == "__main__": pfn, dir_name, file = setup(__file__) r = True models = [ PowerBeutler2017(recon=r, smooth_type="hinton2017", name="Hinton2017"), PowerBeutler2017(recon=r, smooth_type="eh1998", name="EH1998") ] data = PowerSpectrum_SDSS_DR12_Z061_NGC(name="Recon mean", recon=r, min_k=0.02, max_k=0.30) sampler = DynestySampler(temp_dir=dir_name) fitter = Fitter(dir_name) fitter.add_model_and_dataset(models[0], data, name="Hinton2017") fitter.add_model_and_dataset(models[1], data, name="EH1998") fitter.set_sampler(sampler) fitter.set_num_walkers(10) fitter.fit(file) if fitter.should_plot():
import sys sys.path.append("..") from barry.samplers import DynestySampler from barry.cosmology.camb_generator import getCambGenerator from barry.postprocessing import BAOExtractor from barry.config import setup from barry.models import PowerSeo2016, PowerBeutler2017, PowerDing2018, PowerNoda2019 from barry.datasets import PowerSpectrum_SDSS_DR12_Z061_NGC from barry.fitter import Fitter import numpy as np import pandas as pd if __name__ == "__main__": pfn, dir_name, file = setup("../config/pk_individual.py") fitter = Fitter(dir_name, save_dims=2, remove_output=False) c = getCambGenerator() r_s = c.get_data()[0] p = BAOExtractor(r_s) sampler = DynestySampler(temp_dir=dir_name, nlive=200) for r in [True, False]: t = "Recon" if r else "Prerecon" ls = "-" if r else "--" d = PowerSpectrum_SDSS_DR12_Z061_NGC(recon=r, realisation=0) de = PowerSpectrum_SDSS_DR12_Z061_NGC(recon=r, postprocess=p, realisation=0)