SolarRadiationAFNO6H#

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

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