Example #1
0
def test_max_depth():
    model = ctw.create_model(max_depth=0)
    eq_(_get_history_p(model), 1.0)
    model.see_generated([1])
    eq_(model.root.log_p_estim, math.log(0.5))
    eq_(model.root.log_pw, model.root.log_p_estim)

    model.see_generated([1])
    eq_(_get_history_p(model), naive_ctw._estim_kt_p(0, 2))
Example #2
0
def test_max_depth():
    model = ctw.create_model(max_depth=0)
    eq_(_get_history_p(model), 1.0)
    model.see_generated([1])
    eq_(model.root.log_p_estim, math.log(0.5))
    eq_(model.root.log_pw, model.root.log_p_estim)

    model.see_generated([1])
    eq_(_get_history_p(model), naive_ctw._estim_kt_p(0, 2))
Example #3
0
def test_estim_kt_p():
    # The tabulated values are from "Reflections on CTW".
    eq_(ctw._estim_kt_p(0, 1), 0.5)
    eq_(ctw._estim_kt_p(1, 0), 0.5)
    eq_(ctw._estim_kt_p(0, 2), 3 / 8.0)
    eq_(ctw._estim_kt_p(0, 5), 63 / 256.0)

    eq_(ctw._estim_kt_p(3, 1), 5 / 128.0)
    eq_(ctw._estim_kt_p(3, 5), 45 / 32768.0)

    eq_(ctw._estim_kt_p(0, 0), 1.0)
Example #4
0
def test_estim_kt_p():
    # The tabulated values are from "Reflections on CTW".
    eq_(ctw._estim_kt_p(0, 1), 0.5)
    eq_(ctw._estim_kt_p(1, 0), 0.5)
    eq_(ctw._estim_kt_p(0, 2), 3/8.0)
    eq_(ctw._estim_kt_p(0, 5), 63/256.0)

    eq_(ctw._estim_kt_p(3, 1), 5/128.0)
    eq_(ctw._estim_kt_p(3, 5), 45/32768.0)

    eq_(ctw._estim_kt_p(0, 0), 1.0)