pyepo.metric.mse
================

.. py:module:: pyepo.metric.mse

.. autoapi-nested-parse::

   Mean Squared Error



Functions
---------

.. autoapisummary::

   pyepo.metric.mse.MSE


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

.. py:function:: MSE(predmodel: torch.nn.Module, dataloader: torch.utils.data.DataLoader) -> float

   A function to evaluate model performance with MSE

   :param predmodel: a regression neural network for cost prediction
   :param dataloader: Torch dataloader from optDataSet (fields beyond
                      ``(x, c, w, z)`` are ignored)

   :returns: MSE loss
   :rtype: float