Added abstraction to prior definition for more flexible prior forms

coarse-graining / cgnet

learning coarse-grained force fields

BSD 3-Clause "New" or "Revised" License

57 stars 26 forks source link

Added abstraction to prior definition for more flexible prior forms #169

Closed psolsson closed 4 years ago

psolsson commented 4 years ago

Implements _AbstractPriorLayer-class to enable more flexible definition of priors, i.e. without a fixed set of parameters in the CG space during learning and inference. _PriorLayer now inherits forward from _AbstractPriorLayer.

brookehus commented 4 years ago

awesome simon, excited to review this

brookehus commented 4 years ago

@nec4 let's do a 2-approval on this one just cause you're more familiar w/ the priors

nec4 commented 4 years ago

LGTM - all tests pass.