In [1]:
import numpy as np, matplotlib.pyplot as plt, pandas as pd
pd.set_option('display.max_rows', 10)
!date
%load_ext autoreload
%autoreload 2
In [2]:
# if dismod_mr is not installed, it should possible to use
# !conda install --yes pymc
# !pip install dismod_mr
In [3]:
import dismod_mr
In [4]:
from IPython.core.pylabtools import figsize
figsize(11, 5.5)
In [5]:
dm = dismod_mr.data.ModelData()
dm.set_level_bounds('p', lower=0, upper=.1)
dm.setup_model('p')
%time dm.fit(how='mcmc', iter=20_000, burn=10_000, thin=10)
dm.plot()
In [6]:
dm = dismod_mr.data.ModelData()
dm.set_level_bounds('p', lower=0, upper=.1)
dm.set_knots('p', [0,100])
dm.setup_model('p')
%time dm.fit(how='mcmc', iter=20_000, burn=10_000, thin=10)
dm.plot()
In [7]:
dm = dismod_mr.data.ModelData()
dm.set_level_bounds('p', lower=0, upper=1)
dm.set_level_value('p', age_before=1, age_after=100, value=.2)
dm.set_knots('p', [0, 50, 100])
dm.setup_model('p')
%time dm.fit(how='mcmc', iter=20_000, burn=10_000, thin=10)
dm.plot()
In [8]:
dm = dismod_mr.data.ModelData()
dm.set_level_bounds('p', lower=0, upper=1)
dm.set_knots('p', [0, 50, 100])
dm.set_increasing('p', age_start=0, age_end=100) # FIXME: seems like this isn't working
dm.setup_model('p')
%time dm.fit(how='mcmc', iter=20_000, burn=10_000, thin=10)
dm.plot()
In [9]:
!date
In [ ]: