ai2cm / fv3net

explore the FV3 data for parameterization
MIT License
16 stars 3 forks source link

Reservoir separate transforms #2308

Closed AnnaKwa closed 1 year ago

AnnaKwa commented 1 year ago

This PR adds the ability to load separate transforms for input/output/hybrid variable sets (or fit different standard normalization transforms if not specified). These are organized in a new TransformerGroup class that replaces the single autoencoder attribute of the reservoir models.

If the output variable list is different from the input variable list, the training script will use the output transformer when preparing the paired output set.

Added public API:

Refactored public API: