@@ -115,7 +115,7 @@ def sample(draws, step=None, init='auto', n_init=200000, start=None,
115
115
A step function or collection of functions. If there are variables
116
116
without a step methods, step methods for those variables will
117
117
be assigned automatically.
118
- init : str {'ADVI', 'ADVI_MAP', 'MAP', 'NUTS', None}
118
+ init : str {'ADVI', 'ADVI_MAP', 'MAP', 'NUTS', 'auto', None}
119
119
Initialization method to use. Only works for auto-assigned step methods.
120
120
121
121
* ADVI: Run ADVI to estimate starting points and diagonal covariance
@@ -290,15 +290,12 @@ def _sample(draws, step=None, start=None, trace=None, chain=0, tune=None,
290
290
291
291
def iter_sample (draws , step , start = None , trace = None , chain = 0 , tune = None ,
292
292
model = None , random_seed = - 1 ):
293
- """
294
- Generator that returns a trace on each iteration using the given
293
+ """Generator that returns a trace on each iteration using the given
295
294
step method. Multiple step methods supported via compound step
296
295
method returns the amount of time taken.
297
296
298
-
299
297
Parameters
300
298
----------
301
-
302
299
draws : int
303
300
The number of samples to draw
304
301
step : function
0 commit comments