Merge pull request #39 from canyon289/06_ins

Add ArviZ to 06-instructor

Author: ericmjl

notebooks/06-instructor-hierarchical-baseball.ipynb

@@ -13,6 +13,7 @@
     "import theano.tensor as tt\n",
     "from data import load_baseball\n",
     "from utils import despine, ECDF, despine_traceplot\n",
+    "import arviz as az\n",
     "\n",
     "# For deterministic reproducibility.\n",
     "np.random.seed(42)\n",
@@ -160,7 +161,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "traceplot = pm.traceplot(baseline_trace)\n",
+    "traceplot = az.plot_trace(baseline_trace)\n",
     "despine_traceplot(traceplot)"
    ]
   },
@@ -178,7 +179,7 @@
    "outputs": [],
    "source": [
     "ylabels = ylabels = \"AB: \" + df['AB'].astype(str) + ', H: ' + df['H'].astype('str')\n",
-    "pm.forestplot(baseline_trace, ylabels=ylabels, xlim=[0, 1])"
+    "az.plot_forest(baseline_trace, ylabels=ylabels, xlim=[0, 1])"
    ]
   },
   {
@@ -272,7 +273,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "ax_arr = pm.traceplot(trace)\n",
+    "ax_arr = az.plot_trace(trace)\n",
     "despine_traceplot(ax_arr)"
    ]
   },
@@ -290,7 +291,7 @@
    "outputs": [],
    "source": [
     "ylabels = \"AB: \" + df['AB'].astype(str) + ', H: ' + df['H'].astype('str')\n",
-    "pm.forestplot(trace, varnames=['thetas'], ylabels=ylabels, xlim=[0, 1])"
+    "az.plot_trace(trace, var_names=['thetas'], ylabels=ylabels, xlim=[0, 1])"
    ]
   },
   {
@@ -376,9 +377,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "bayesian-modelling-tutorial",
+   "display_name": "Python 3",
    "language": "python",
-   "name": "bayesian-modelling-tutorial"
+   "name": "python3"
   },
   "language_info": {
    "codemirror_mode": {
@@ -390,7 +391,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.6"
+   "version": "3.6.8"
   }
  },
  "nbformat": 4,