earth2studio.run.ensemble#
- earth2studio.run.ensemble(time, nsteps, nensemble, prognostic, data, io, perturbation_method, batch_size=None, output_coords={}, device=None)[source]#
Ensemble workflow
- Parameters:
time (list[str] | list[datetime] | list[np.datetime64]) – List of string, datetimes or np.datetime64
nsteps (int) – Number of forecast steps
nensemble (int) – Number of ensemble members to run inference for.
prognostic (PrognosticModel) – Prognostic models
data (DataSource) – Data source
io (IOBackend) – IO object
perturbation_method (PerturbationMethod) – Method to perturb the initial condition to create an ensemble.
batch_size (Optional[int], optional) – Number of ensemble members to run in a single batch, by default None.
device (Optional[torch.device], optional) – Device to run inference on, by default None
output_coords (OrderedDict[str, ndarray])
- Returns:
Output IO object
- Return type:
IOBackend
