rollout_ensemble_bred_vector

rollout_ensemble_bred_vector#

Attributes#

logger

description

Functions#

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

Compute metrics and update metrics_results.

calculate_ensemble_metrics(ensemble_preds, true_values)

Calculate RMSE and STD per channel

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

Module Contents#

rollout_ensemble_bred_vector.logger#
rollout_ensemble_bred_vector.compute_metrics(metrics, y_pred, y, date_time, forecast_step, utc_datetime)#

Compute metrics and update metrics_results.

rollout_ensemble_bred_vector.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_ensemble_bred_vector.predict(rank, world_size, conf, backend=None, p=None)#
rollout_ensemble_bred_vector.description = 'Rollout AI-NWP forecasts'#
Contents