earth2studio.models.dx.SolarRadiationAFNO6H#

class earth2studio.models.dx.SolarRadiationAFNO6H(core_model, freq, era5_mean, era5_std, ssrd_mean, ssrd_std, orography, landsea_mask, sincos_latlon)[source]#

Solar Radiation AFNO diagnostic model for 6-hour frequency predictions. Predicts the accumulated global surface solar radiation over 6 hours [Jm^-2].

Parameters:
  • core_model (Module)

  • freq (str)

  • era5_mean (Tensor)

  • era5_std (Tensor)

  • ssrd_mean (Tensor)

  • ssrd_std (Tensor)

  • orography (Tensor)

  • landsea_mask (Tensor)

  • sincos_latlon (Tensor)

__call__(x, coords)#

Forward pass of diagnostic

Parameters:
  • x (Tensor)

  • coords (OrderedDict[str, ndarray])

Return type:

tuple[Tensor, OrderedDict[str, ndarray]]

classmethod load_default_package()[source]#

Load prognostic package

Return type:

Package

classmethod load_model(package)#

Load prognostic from package

Parameters:

package (Package) – Package to load model from

Returns:

Prognostic model

Return type:

PrognosticModel