:mod:`fwi_forecast`
===================

.. py:module:: fwi_forecast

.. autoapi-nested-parse::

   The dataset class to be used with fwi-forcings and fwi-forecast data.



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

.. py:class:: ModelDataset(dates, forecast_dir, hparams=None, **kwargs)

   Bases: :class:`dataloader.base_loader.ModelDataset`

   .. autoapi-inheritance-diagram:: fwi_forecast.ModelDataset
      :parts: 1

   The dataset class responsible for loading the data and providing the samples for
   training.

   :param BaseDataset: Base Dataset class to inherit from
   :type BaseDataset: base_loader.BaseDataset