Skip to content

add a prediction function and method to handle forward simulations #286

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
LegrandNico opened this issue Mar 31, 2025 · 1 comment
Open
Assignees
Labels
enhancement New feature or request networks

Comments

@LegrandNico
Copy link
Collaborator

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 LegrandNico added enhancement New feature or request networks labels Mar 31, 2025
@LegrandNico 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
@LegrandNico
Copy link
Collaborator Author

@SylvainEstebe

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request networks
Projects
None yet
Development

No branches or pull requests

2 participants