Trainer

flood_forecast.trainer.train_function(model_type: str, params: Dict)

Function to train a Model(TimeSeriesModel) or da_rnn. Will return the trained model :param model_type: Type of the model. In almost all cases this will be ‘PyTorch’ :type model_type: str :param params: Dictionary containing all the parameters needed to run the model :type Dict:

flood_forecast.trainer.main()

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