Exemplos de ADVI.run_profiling em Python

Exemplo n.º 1

Arquivo: test_variational_inference.py Projeto: zaxtax/pymc3

def test_remove_scan_op():
    with pm.Model():
        pm.Normal('n', 0, 1)
        inference = ADVI()
        buff = six.StringIO()
        inference.run_profiling(n=10).summary(buff)
        assert 'theano.scan_module.scan_op.Scan' not in buff.getvalue()
        buff.close()

Exemplo n.º 2

Arquivo: test_variational_inference.py Projeto: pandumagdum/pymc3

def test_remove_scan_op():
    with pm.Model():
        pm.Normal('n', 0, 1)
        inference = ADVI()
        buff = io.StringIO()
        inference.run_profiling(n=10).summary(buff)
        assert 'theano.scan_module.scan_op.Scan' not in buff.getvalue()
        buff.close()

Exemplo n.º 3

Arquivo: test_variational_inference.py Projeto: walking-appa/pymc3

def test_remove_scan_op():
    with pm.Model():
        pm.Normal("n", 0, 1)
        inference = ADVI()
        buff = io.StringIO()
        inference.run_profiling(n=10).summary(buff)
        assert "aesara.scan.op.Scan" not in buff.getvalue()
        buff.close()