diagnostic#
- earth2studio.run.diagnostic(
- time,
- nsteps,
- prognostic,
- diagnostic,
- data,
- io,
- output_coords={},
- device=None,
- verbose=True,
Built in diagnostic workflow. This workflow creates a determinstic inference pipeline that couples a prognostic model with a diagnostic model.
- Parameters:
time (list[str] | list[datetime] | list[np.datetime64]) – List of string, datetimes or np.datetime64
nsteps (int) – Number of forecast steps
prognostic (PrognosticModel) – Prognostic model
diagnostic (DiagnosticModel) – Diagnostic model, must be on same coordinate axis as prognostic
data (DataSource | ForecastSource) – Data source
io (IOBackend) – IO object
output_coords (CoordSystem, optional) – IO output coordinate system override, by default OrderedDict({})
device (torch.device, optional) – Device to run inference on, by default None
verbose (bool, optional) – Print inference progress, by default True
- Returns:
Output IO object
- Return type:
IOBackend
