run.diagnostic

Import path: earth2studio.run.diagnostic

View source on GitHub

Documentation

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[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, default: OrderedDict({}) ) –

  IO output coordinate system override, by default OrderedDict({})

• device (device, default: None ) –

  Device to run inference on, by default None

• verbose (bool, default: True ) –

  Print inference progress, by default True

• checkpoint (Checkpoint, default: NullCheckpoint() ) –

  Checkpoint manager or checkpoint session used to record and resume workflow progress, by default no checkpoint

Returns:

• IOBackend –

  Output IO object

Examples using earth2studio.run.diagnostic

Running Diagnostic InferenceBasic prognostic + diagnostic inference workflow. Extending Diagnostic ModelsImplementing a custom diagnostic model