You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
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.
The text was updated successfully, but these errors were encountered:
LegrandNico
changed the title
add a prediction function and mathod to handle forward simulations
add a prediction function and method to handle forward simulations
Mar 31, 2025
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:
Where
predictions_trajectories
should be a dictionary of arrays or size (n_predictions, time_steps) each (i.e. just likenode_trajectories
but for multiple simulations).The
predict
method should handle the creation of the scan function, and the vectorisation of random keys.The text was updated successfully, but these errors were encountered: