🎨 Update plots from pymc to arviz dependencies & replace param sd with sigma

Author: CloudChaoszero

examples/case_studies/disaster_model.py

@@ -8,7 +8,7 @@
 
 """
 
-
+import arviz as az
 import theano.tensor as tt
 
 from numpy import arange, array
@@ -152,4 +152,4 @@
     start = {"early_mean": 2.0, "late_mean": 3.0}
 
     tr = pm.sample(1000, tune=500, start=start)
-    pm.traceplot(tr)
+    pm.plot_trace(tr)

examples/case_studies/disaster_model_theano_op.py

@@ -4,7 +4,7 @@
 Note that gradient based samplers will not work.
 """
 
-
+import arviz as az
 import theano.tensor as tt
 
 from numpy import arange, array, empty
@@ -166,4 +166,4 @@ def rate_(switchpoint, early_mean, late_mean):
     start = {"early_mean": 2.0, "late_mean": 3.0}
 
     tr = pm.sample(1000, tune=500, start=start, step=[step1, step2], cores=2)
-    pm.traceplot(tr)
+    az.plot_trace(tr)

examples/case_studies/garch_example.py

@@ -1,7 +1,8 @@
 import numpy as np
 import theano.tensor as tt
 
-from pymc3 import Model, Normal, Uniform, sample, summary
+from arviz import summary
+from pymc3 import Model, Normal, Uniform, sample
 
 """
 Example from Stan - slightly altered

examples/case_studies/gelman_schools.py

@@ -1,6 +1,7 @@
 import numpy as np
 
-from pymc3 import HalfCauchy, Model, Normal, loo, sample
+from arviz import loo
+from pymc3 import HalfCauchy, Model, Normal, sample
 
 """Original Stan model
 

examples/case_studies/lightspeed_example.py

@@ -1,5 +1,5 @@
 import numpy as np
-
+import arviz as az
 import pymc3 as pm
 
 light_speed = np.array(
@@ -92,7 +92,7 @@ def run(n=5000):
     with model_1:
         trace = pm.sample(n)
 
-        pm.summary(trace)
+        az.summary(trace)
 
 
 if __name__ == "__main__":

examples/pymc3_howto/LKJ_correlation.py

@@ -1,6 +1,6 @@
 import numpy as np
 import theano.tensor as tt
-
+import arviz as az
 from numpy.random import multivariate_normal
 
 import pymc3 as pm
@@ -53,7 +53,7 @@ def run(n=1000):
         n = 50
     with model:
         trace = pm.sample(n)
-    pm.traceplot(
+    az.plot_trace(
         trace, varnames=["mu", "r"], lines={"mu": mu_r, "r": corr_r[np.triu_indices(n_var, k=1)]}
     )
 

examples/pymc3_howto/rankdata_ordered.py

@@ -46,14 +46,14 @@ def run(n=1500):
     with m:
         trace = pm.sample(n)
 
-    pm.traceplot(trace, varnames=["mu_hat"])
+    az.plot_trace(trace, varnames=["mu_hat"])
 
     print("Example observed data: ")
     print(y[:30, :].T)
     print("The true ranking is: ")
     print(yreal.flatten())
     print("The Latent mean is: ")
-    latentmu = np.hstack(([0], pm.summary(trace, varnames=["mu_hat"])["mean"].values))
+    latentmu = np.hstack(([0], az.summary(trace, varnames=["mu_hat"])["mean"].values))
     print(np.round(latentmu, 2))
     print("The estimated ranking is: ")
     print(np.argsort(latentmu))

examples/samplers/samplers_mvnormal.py

@@ -9,6 +9,7 @@
 
 import time
 
+import arviz as az
 import numpy as np
 import pandas as pd
 import theano.tensor as tt
@@ -50,7 +51,7 @@ def run(steppers, p):
             runtimes[name] = time.time() - t_start
             print("{} samples across {} chains".format(len(mt) * mt.nchains, mt.nchains))
             traces[name] = mt
-            en = pm.ess(mt)
+            en = az.ess(mt)
             print(f"effective: {en}\r\n")
             if USE_XY:
                 effn[name] = np.mean(en["x"]) / len(mt) / mt.nchains

examples/time_series/arma_example.py

@@ -2,6 +2,7 @@
 
 from theano import scan, shared
 
+import arviz as az
 import pymc3 as pm
 
 """
@@ -82,8 +83,8 @@ def run(n_samples=1000):
     with model:
         trace = pm.sample(draws=n_samples, tune=1000, target_accept=0.99)
 
-    pm.plots.traceplot(trace)
-    pm.plots.forestplot(trace)
+    az.plot_trace(trace)
+    az.plot_forest(trace)
 
 
 if __name__ == "__main__":