pyhgf.utils.sample#

pyhgf.utils.sample(network, time_steps, n_predictions=1, rng_key=Array((), dtype=key<fry>) overlaying: [0 0])[source]#

Sample n_predictions forward in generative mode.

Parameters:
  • network (Network) – The network instance used for generating predictions.

  • time_steps (Array | ndarray | bool | number | bool | int | float | complex) – Array of time steps.

  • n_predictions (int) – Number of predictions to generate. Defaults to 1.

  • rng_key (Array | ndarray | bool | number | bool | int | float | complex) – Random number generator key, by default PRNGKey(0).

Returns:

Dictionary of predictions for each parameter/node.

Return type:

predictions