predict404#

Attributes#

Functions#

predict(rank, world_size, conf, dataset)

Loads model defined in configuration object and applies it to

Module Contents#

predict404.logger#
predict404.predict(rank, world_size, conf, dataset)#

Loads model defined in configuration object and applies it to data in dataset (using test split defined in conf). Returns a list of xarray objects, one for each predicted output.

predict404.description = 'Rollout AI-NWP forecasts'#