Skip to content

Commit a223fdf

Browse files
authored
Merge pull request #33 from canyon289/03_instructor
Add Arviz to 03-instructor
2 parents 151d208 + 6d10355 commit a223fdf

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

notebooks/03-instructor-two-group-iq.ipynb

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
"import matplotlib.pyplot as plt\n",
1212
"import pymc3 as pm\n",
1313
"import seaborn as sns\n",
14+
"import arviz as az\n",
1415
"from utils import ECDF\n",
1516
"from data import load_kruschke\n",
1617
"\n",
@@ -290,7 +291,7 @@
290291
"outputs": [],
291292
"source": [
292293
"# Use `pm.plot_posterior()`\n",
293-
"pm.plot_posterior(trace, varnames=['mu_drug', 'mu_placebo'])"
294+
"az.plot_posterior(trace, var_names=['mu_drug', 'mu_placebo'])"
294295
]
295296
},
296297
{
@@ -377,7 +378,7 @@
377378
"metadata": {},
378379
"outputs": [],
379380
"source": [
380-
"pm.plot_posterior(trace=trace, varnames=['diff_means', 'effect_size'])"
381+
"az.plot_posterior(trace=trace, var_names=['diff_means', 'effect_size'])"
381382
]
382383
},
383384
{
@@ -488,7 +489,7 @@
488489
"metadata": {},
489490
"outputs": [],
490491
"source": [
491-
"traces = pm.traceplot(trace_alt)"
492+
"traces = az.traceplot(trace_alt)"
492493
]
493494
},
494495
{
@@ -497,7 +498,7 @@
497498
"metadata": {},
498499
"outputs": [],
499500
"source": [
500-
"pm.plot_posterior(trace_alt, varnames=['mu', 'effect_size'])"
501+
"az.plot_posterior(trace_alt, var_names=['mu', 'effect_size'])"
501502
]
502503
},
503504
{
@@ -521,9 +522,9 @@
521522
],
522523
"metadata": {
523524
"kernelspec": {
524-
"display_name": "bayesian-modelling-tutorial",
525+
"display_name": "Python 3",
525526
"language": "python",
526-
"name": "bayesian-modelling-tutorial"
527+
"name": "python3"
527528
},
528529
"language_info": {
529530
"codemirror_mode": {
@@ -535,7 +536,7 @@
535536
"name": "python",
536537
"nbconvert_exporter": "python",
537538
"pygments_lexer": "ipython3",
538-
"version": "3.6.6"
539+
"version": "3.6.8"
539540
},
540541
"toc": {
541542
"base_numbering": 1,

0 commit comments

Comments
 (0)