Skip to content

Commit 151d208

Browse files
authored
Merge pull request #34 from canyon289/03_student
Add ArviZ to 03-student
2 parents c47d68c + 5405b31 commit 151d208

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

notebooks/03-student-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",
@@ -287,7 +288,7 @@
287288
"metadata": {},
288289
"outputs": [],
289290
"source": [
290-
"# Use `pm.plot_posterior()`\n",
291+
"# Use `az.plot_posterior()`\n",
291292
"pm.__________________"
292293
]
293294
},
@@ -376,7 +377,7 @@
376377
"metadata": {},
377378
"outputs": [],
378379
"source": [
379-
"# Use `pm.plot_posterior`, but pass in different varnames\n",
380+
"# Use `az.plot_posterior`, but pass in different varnames\n",
380381
"_______________________"
381382
]
382383
},
@@ -488,7 +489,7 @@
488489
"metadata": {},
489490
"outputs": [],
490491
"source": [
491-
"traces = pm.traceplot(trace_alt)"
492+
"traces = az.plot_trace(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)