diff --git a/pyro/contrib/epidemiology/compartmental.py b/pyro/contrib/epidemiology/compartmental.py index 50ef566fc4..8d5003ea02 100644 --- a/pyro/contrib/epidemiology/compartmental.py +++ b/pyro/contrib/epidemiology/compartmental.py @@ -182,10 +182,10 @@ def transition(self, params, state, t): Note that this method is called under multiple different interpretations, including batched and vectorized interpretations. During :meth:`generate` this is called to generate a single sample. - During :meth:`heuristic` this is called to generate a batch of sample + During :meth:`heuristic` this is called to generate a batch of samples for SMC. During :meth:`fit` this is called both in vectorized form - (vectorizing over time) and insequential form (for a single time step); - both forms enumerate over discrete latent variables. During + (vectorizing over time) and in sequential form (for a single time + step); both forms enumerate over discrete latent variables. During :meth:`predict` this is called to forecast a batch of samples, conditioned on posterior samples for the time interval ``[0:duration]``. diff --git a/pyro/contrib/epidemiology/models.py b/pyro/contrib/epidemiology/models.py index 2672ad663e..e2b2b1fe68 100644 --- a/pyro/contrib/epidemiology/models.py +++ b/pyro/contrib/epidemiology/models.py @@ -566,7 +566,7 @@ class SparseSIRModel(CompartmentalModel): This model demonstrates how to implement a custom :meth:`compute_flows` method. A custom method is needed in this model because inhabitants of the - ``S`` compartment can transition to both the ``I`` and ``o`` compartments, + ``S`` compartment can transition to both the ``I`` and ``O`` compartments, allowing duplication. :param int population: Total ``population = S + I + R``.