def test_parallel_tempering(): rng = irm.RNG() D1_N = 100 model_name = "BetaBernoulliNonConj" d = {'domains' : {'d1' : {'N' : D1_N}}, 'relations' : {'R1' : {'relation' : ('d1', 'd1'), 'model' : model_name}}} l = {} new_latent, new_data = data.synth.prior_generate(l, d) config = [('parallel_tempering', {'temps' : [1.0, 2.0, 4.0, 8.0], 'subkernels' : runner.default_kernel_nonconj_config()})] r = runner.Runner(new_latent, new_data, config) for i in range(100): print "tt", i, r.get_score() r.run_iters(1)
from irm import irmio, runner from irm import relation import util as putil SAMPLE_SETS = 20 SAMPLES_N = 1000 ITERS_PER_SAMPLE = 10 # SAMPLE_SETS = 4 # SAMPLES_N = 100 # ITERS_PER_SAMPLE = 2 SEEDS = 1 MODEL_CLASSES = ['conj', 'nonconj'] tt_config_nonconj = [('parallel_tempering', {'temps' : [1.0, 2.0, 4.0, 8.0], 'subkernels' : runner.default_kernel_nonconj_config()})] tt_config_conj = [('parallel_tempering', {'temps' : [1.0, 2.0, 4.0, 8.0], 'subkernels' : runner.default_kernel_config()})] KERNEL_CONFIGS = {'default' : {'conj' : runner.default_kernel_config(), 'nonconj' : runner.default_kernel_nonconj_config()}} def t1_t2_datasets(): T1_N = 8 T2_N = 7 for mc in MODEL_CLASSES: for seed in range(SEEDS): filename_base = "srm.t1xt2.%d.%d.%d.%s" % (T1_N, T2_N, seed, mc) latent_filename = filename_base + ".latent"
from irm import relation import util as putil SAMPLE_SETS = 20 SAMPLES_N = 1000 ITERS_PER_SAMPLE = 10 # SAMPLE_SETS = 4 # SAMPLES_N = 100 # ITERS_PER_SAMPLE = 2 SEEDS = 1 MODEL_CLASSES = ['conj', 'nonconj'] tt_config_nonconj = [('parallel_tempering', { 'temps': [1.0, 2.0, 4.0, 8.0], 'subkernels': runner.default_kernel_nonconj_config() })] tt_config_conj = [('parallel_tempering', { 'temps': [1.0, 2.0, 4.0, 8.0], 'subkernels': runner.default_kernel_config() })] KERNEL_CONFIGS = { 'default': { 'conj': runner.default_kernel_config(), 'nonconj': runner.default_kernel_nonconj_config() } }