Python GeometricModel示例

示例#1

def test_geometric_model_stats():
    model = GeometricModel(alpha=4, beta=6)

    assert model.pppdf(0) == approx(0)
    assert model.pppdf(4) == approx(0.0783216783)
    assert model.ppmean() == approx(3)
    assert model.ppvar() == approx(3)

示例#2

def test_geometric_model_update():
    model = GeometricModel(alpha=1, beta=1)
    data = np.array([1, 2, 4, 2, 3, 3, 5, 4, 1, 4])

    model.update(data)

    assert model.n_samples_ == 10

示例#3

文件： test_binomial.py 项目： guillermo-navas-palencia/cprior

def test_binomial_ab_check_models():
    modelA = BinomialModel(m=10, alpha=1, beta=1)
    modelB = GeometricModel(alpha=1, beta=1)

    with raises(TypeError):
        BinomialABTest(modelA=modelA, modelB=modelB)

示例#4

def test_geometric_mv_check_model_input():
    modelA = GeometricModel(alpha=1, beta=1)
    modelB = GeometricModel(alpha=1, beta=1)

    with raises(TypeError):
        GeometricMVTest(models=[modelA, modelB])

示例#5

def test_geometric_ab_check_models():
    modelA = BernoulliModel(alpha=1, beta=1)
    modelB = GeometricModel(alpha=1, beta=1)

    with raises(TypeError):
        GeometricABTest(modelA=modelA, modelB=modelB)

示例#6

def test_geometric_model_mean_var():
    model = GeometricModel(alpha=1, beta=6)

    assert np.isnan(model.ppmean())
    assert np.isnan(model.ppvar())

示例#7

def test_geometric_pppdf_x():
    model = GeometricModel(alpha=4, beta=6)

    assert model.pppdf([-1, 0, 1, 10]) == approx([0, 0, 0.4, 0.0086687306])