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()
}
}
