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)

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()

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