Trainer

flood_forecast.trainer.train_function(model_type: str, params: Dict[KT, VT])

Function to train a Model(TimeSeriesModel) or da_rnn. Will return the trained model model_type str: Type of the model (for now) must be da_rnn or :params dict: Dictionary containing all the parameters needed to run the model

flood_forecast.trainer.main()

Main function which is called from the command line. Entrypoint for all ML models.