rollout_metrics_noisy_ic

rollout_metrics_noisy_ic#

Attributes#

logger

description

Functions#

compute_metrics(metrics, y_pred, y, date_time, ...)

calculate_ensemble_metrics(ensemble_preds, true_values)

Calculate RMSE and STD per channel

predict(rank, world_size, conf[, backend, p])

Module Contents#

rollout_metrics_noisy_ic.logger#
rollout_metrics_noisy_ic.compute_metrics(metrics, y_pred, y, date_time, forecast_step, utc_datetime)#
rollout_metrics_noisy_ic.calculate_ensemble_metrics(ensemble_preds, true_values)#

Calculate RMSE and STD per channel

Parameters:

  • ensemble_preds – (n_members, batch, channels, 1, height, width)

  • true_values – (1, batch, channels, 1, height, width)

rollout_metrics_noisy_ic.predict(rank, world_size, conf, backend=None, p=None)#
rollout_metrics_noisy_ic.description = 'Rollout AI-NWP forecasts'#