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

The network instance used for generating predictions.

time_steps

Array of time steps.

n_predictions

Number of predictions to generate. Defaults to 1.

rng_key

Random number generator key, by default PRNGKey(0).

Returns:
predictions

Dictionary of predictions for each parameter/node.

Parameters:
  • network (Network)

  • time_steps (Array | ndarray | bool | number | bool | int | float | complex)

  • n_predictions (int)

  • rng_key (Array | ndarray | bool | number | bool | int | float | complex)

Return type:

Dict[int | str, Dict]