Skip to content

Commit e808939

Browse files
committed
make approximation docs more verbose
1 parent bc56865 commit e808939

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

pymc3/variational/approximations.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@ class MeanField(Approximation):
2525
2626
Parameters
2727
----------
28-
local_rv : dict
29-
mapping {model_variable -> local_variable}
28+
local_rv : dict[var->tuple]
29+
mapping {model_variable -> local_variable (:math:`\\mu`, math:`\\rho`)}
3030
Local Vars are used for Autoencoding Variational Bayes
3131
See (AEVB; Kingma and Welling, 2014) for details
3232
@@ -100,8 +100,8 @@ class FullRank(Approximation):
100100
101101
Parameters
102102
----------
103-
local_rv : dict
104-
mapping {model_variable -> local_variable}
103+
local_rv : dict[var->tuple]
104+
mapping {model_variable -> local_variable (:math:`\\mu`, math:`\\rho`)}
105105
Local Vars are used for Autoencoding Variational Bayes
106106
See (AEVB; Kingma and Welling, 2014) for details
107107
@@ -240,9 +240,9 @@ class Empirical(Approximation):
240240
Parameters
241241
----------
242242
trace : MultiTrace
243-
local_rv : dict
244-
Experimental for Histogram approximation
245-
mapping {model_variable -> local_variable}
243+
local_rv : dict[var->tuple]
244+
Experimental for Empirical Distribution
245+
mapping {model_variable -> local_variable (:math:`\\mu`, math:`\\rho`)}
246246
Local Vars are used for Autoencoding Variational Bayes
247247
See (AEVB; Kingma and Welling, 2014) for details
248248

0 commit comments

Comments
 (0)