def get_final_stage(homepath, n_stages, model): """ Combine Metropolis results into final stage to get one single chain for plotting results. """ util.ensuredir(homepath) mtraces = [] for stage in range(n_stages): logger.info('Loading Metropolis stage %i' % stage) stage_outpath = os.path.join(homepath, 'stage_%i' % stage) mtraces.append( backend.load( name=stage_outpath, model=model)) ctrace = backend.concatenate_traces(mtraces) outname = os.path.join(homepath, 'stage_final') if os.path.exists(outname): logger.info('Removing existing previous final stage!') shutil.rmtree(outname) util.ensuredir(outname) logger.info('Creating final Metropolis stage') text.dump(name=outname, trace=ctrace)
def setUpClass(cls): super(TestTextDumpFunction, cls).setUpClass() text.dump(cls.name1, cls.mtrace1) with cls.model: cls.mtrace1 = text.load(cls.name1)
def setup_class(cls): super().setup_class() text.dump(cls.name1, cls.mtrace1) with cls.model: cls.mtrace1 = text.load(cls.name1)