rollout_metrics_noisy_ic
========================

.. py:module:: rollout_metrics_noisy_ic


Attributes
----------

.. autoapisummary::

   rollout_metrics_noisy_ic.logger
   rollout_metrics_noisy_ic.description


Functions
---------

.. autoapisummary::

   rollout_metrics_noisy_ic.compute_metrics
   rollout_metrics_noisy_ic.calculate_ensemble_metrics
   rollout_metrics_noisy_ic.predict


Module Contents
---------------

.. py:data:: logger

.. py:function:: compute_metrics(metrics, y_pred, y, date_time, forecast_step, utc_datetime)

.. py:function:: calculate_ensemble_metrics(ensemble_preds, true_values)

   Calculate RMSE and STD per channel

   :param ensemble_preds: (n_members, batch, channels, 1, height, width)
   :param true_values: (1, batch, channels, 1, height, width)


.. py:function:: predict(rank, world_size, conf, backend=None, p=None)

.. py:data:: description
   :value: 'Rollout AI-NWP forecasts'