nbren12 / uwnet

Neural Networks based unified physics parameterization for atmospheric models
MIT License
3 stars 2 forks source link

Improve interface routines #6

Closed nbren12 closed 6 years ago

nbren12 commented 6 years ago

I need functions that make it easy to call the neural network with

To do this, I should follow a hub and spoke model. The hub will be a common data-structure (e.g. dict of torch arrays) and the spokes will convert data-structures to this central data structure. Then, I should only need to write one interface routine for the central data structure, and then provide translation routines from xarrays to this data structure.

nbren12 commented 6 years ago

done. uwnet.interface offers interfaces for numpy, xarray, and torch.