def test_runner_specify_hp_kernels(): N, V = 10, 20 defn = model_definition(N, V) data = toy_dataset(defn) view = data prng = rng() latent = model.initialize(defn, view, prng) kernels = ['crf'] + \ runner.second_dp_hp_kernel_config(defn) + \ runner.base_dp_hp_kernel_config(defn) r = runner.runner(defn, view, latent, kernels) r.run(prng, 1)
def test_runner_base_dp_valid(): N, V = 10, 20 defn = model_definition(N, V) data = toy_dataset(defn) prng = rng() latent = model.initialize(defn, data, prng) old_beta = latent.beta old_alpha = latent.alpha kernels = ['crf'] + \ runner.base_dp_hp_kernel_config(defn) r = runner.runner(defn, data, latent, kernels) r.run(prng, 10) assert_almost_equals(latent.beta, old_beta) assert_almost_equals(latent.alpha, old_alpha) assert latent.gamma > 0