From 9b55807be3db846ae89c31af42fe14fb35723a33 Mon Sep 17 00:00:00 2001 From: Edoardo Balzani Date: Tue, 31 Oct 2023 18:16:08 -0400 Subject: [PATCH 1/2] Update src/neurostatslib/noise_model.py Co-authored-by: William F. Broderick --- src/neurostatslib/noise_model.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/neurostatslib/noise_model.py b/src/neurostatslib/noise_model.py index 8be1be24..7310d468 100644 --- a/src/neurostatslib/noise_model.py +++ b/src/neurostatslib/noise_model.py @@ -124,9 +124,8 @@ def emission_probability( Returns ------- - jnp.ndarray - Random numbers generated from the desired distribution based on the `predicted_rate` scale parameter - if needed. + : + Random numbers generated from the noise model with `predicted_rate`. """ pass From bbe8213904f32e24ce44131ae6e633c695e72cf1 Mon Sep 17 00:00:00 2001 From: Edoardo Balzani Date: Tue, 31 Oct 2023 18:18:16 -0400 Subject: [PATCH 2/2] Update src/neurostatslib/noise_model.py Co-authored-by: William F. Broderick --- src/neurostatslib/noise_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/neurostatslib/noise_model.py b/src/neurostatslib/noise_model.py index 7310d468..6401b54a 100644 --- a/src/neurostatslib/noise_model.py +++ b/src/neurostatslib/noise_model.py @@ -131,7 +131,7 @@ def emission_probability( @abc.abstractmethod def residual_deviance(self, predicted_rate: jnp.ndarray, spike_counts: jnp.ndarray): - r"""Compute the residual deviance for a Poisson model. + r"""Compute the residual deviance for the noise model. Parameters ----------