better docs

Author: mitzimorris

docsrc/examples/MCMC Sampling.ipynb

@@ -316,7 +316,7 @@
    },
    "outputs": [],
    "source": [
-    "fit.summary()"
+    "eight_schools_fit.summary()"
    ]
   },
   {
@@ -344,6 +344,15 @@
     "## Accessing the sampler outputs"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "fit = model.sample(data='bernoulli.data.json')"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -375,7 +384,10 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "### Extracting the draws as structured Stan program variables"
+    "### Extracting the draws as structured Stan program variables\n",
+    "\n",
+    "Per-variable draws can be accessed as either a numpy.ndarray object\n",
+    "via method `stan_variable` or as an xarray.Dataset object via `draws_xr`."
    ]
   },
   {
@@ -384,8 +396,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "for k, v in fit.stan_variables().items():\n",
-    "    print(f'name: {k}, shape: {v.shape}')"
+    "fit.stan_variable('theta')"
    ]
   },
   {
@@ -401,7 +412,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "### Extracting sampler method diagnostics"
+    "The `stan_variables` method returns a Python `dict` over all Stan variables in the output."
    ]
   },
   {
@@ -410,17 +421,25 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "for k, v in fit.method_variables().items():\n",
+    "for k, v in fit.stan_variables().items():\n",
     "    print(f'name: {k}, shape: {v.shape}')"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "### Extracting sampler method diagnostics"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
-    "print(f'divergences per chain?\\n{fit.divergences}\\niterations at maxtreedepth per chain?\\n{fit.max_treedepths}')"
+    "for k, v in fit.method_variables().items():\n",
+    "    print(f'name: {k}, shape: {v.shape}')"
    ]
   },
   {