pyhgf icon indicating copy to clipboard operation
pyhgf copied to clipboard

add a prediction function and method to handle forward simulations

Open LegrandNico opened this issue 9 months ago • 1 comments

Now that #274 is merged, we need to implement the corresponding scan function so the main Network class can sample predictions when provided with time step sequences.

The main call should look like this:

predictions_trajectories = Network.predict(n_predictions=5, time_steps=np.ones(20))

Where predictions_trajectories should be a dictionary of arrays or size (n_predictions, time_steps) each (i.e. just like node_trajectories but for multiple simulations).

The predict method should handle the creation of the scan function, and the vectorisation of random keys.

LegrandNico avatar Mar 31 '25 14:03 LegrandNico

@SylvainEstebe

LegrandNico avatar Apr 01 '25 07:04 LegrandNico