Merge pull request #958 from x2apps/ElemwiseCategorical-var2vars

Chris Fonnesbeck · Chris Fonnesbeck · commit 55dd7bb79876 · 2016-02-03T11:07:49.000-06:00
Update ElemwiseCategorical with vars=[category]
diff --git a/pymc3/examples/gaussian_mixture_model.ipynb b/pymc3/examples/gaussian_mixture_model.ipynb
@@ -147,7 +147,7 @@
     "# fit model\n",
     "with model:\n",
     "    step1 = pm.Metropolis(vars=[p, sd, means])\n",
-    "    step2 = pm.ElemwiseCategoricalStep(var=category, values=[0, 1, 2])\n",
+    "    step2 = pm.ElemwiseCategoricalStep(vars=[category], values=[0, 1, 2])\n",
     "    tr = pm.sample(10000, step=[step1, step2])"
    ]
   },

Original file line number	Diff line number	Diff line change
`@@ -147,7 +147,7 @@`
`147`	`147`	`"# fit model\n",`
`148`	`148`	`"with model:\n",`
`149`	`149`	`" step1 = pm.Metropolis(vars=[p, sd, means])\n",`
`150`		`- " step2 = pm.ElemwiseCategoricalStep(var=category, values=[0, 1, 2])\n",`
	`150`	`+ " step2 = pm.ElemwiseCategoricalStep(vars=[category], values=[0, 1, 2])\n",`
`151`	`151`	`" tr = pm.sample(10000, step=[step1, step2])"`
`152`	`152`	`]`
`153`	`153`	`},`