Extensions of QRNN code.

atmtools / typhon

Tools for atmospheric research

http://www.radiativetransfer.org/

MIT License

58 stars 33 forks source link

Extensions of QRNN code. #369

Closed simonpf closed 4 years ago

simonpf commented 4 years ago

This extends the current implementation of QRNN to support both Pytorch and Keras as backends. The QRNN class has been refactored to separate general QRNN functionality from the underlying neural network model. Some functionality has been added e.g. computing the posterior mean and sampling from a fitted Gaussian.

This work is part of an upcoming study on cloud correction at 183 GHz.

simonpf commented 4 years ago

Thank you for the suggestions! Fixed!