def next(self): ret = ParallelTemperingSampler.next(self) if self.nsamples in self.adapt_at: ## TODO: Maybe make this faster? self.adapt_temperatures() return ret
def __init__(self, make_h0, data, adapt_at=[50000, 100000, 200000, 300000, 500000, 1000000], **kwargs): ParallelTemperingSampler.__init__(self, make_h0, data, **kwargs) self.adapt_at = adapt_at