Source code for flood_forecast.da_rnn.custom_types

import collections
import typing

import numpy as np




class TrainConfig(typing.NamedTuple):
    T: int
    train_size: int
    batch_size: int
    loss_func: typing.Callable





class TrainData(typing.NamedTuple):
    feats: np.ndarray
    targs: np.ndarray



DaRnnNet = collections.namedtuple("DaRnnNet", ["encoder", "decoder", "enc_opt", "dec_opt"])