nemos.observation_models.Observations.sample_generator#

abstractmethod Observations.sample_generator(key, predicted_rate, scale=1.0)[source]#

Sample from the estimated distribution.

This method generates random numbers from the desired distribution based on the given predicted_rate.

Parameters:

  • key (Array) – Random key used for the generation of random numbers in JAX.

  • predicted_rate (Array) – Expected rate of the distribution. Shape (n_time_bins, ), or (n_time_bins, n_neurons)..

  • scale (Union[float, Array]) – Scale parameter for the distribution.

Return type:

Array

Returns:

Random numbers generated from the observation model with predicted_rate.