graphium icon indicating copy to clipboard operation
graphium copied to clipboard

Replace the `pre-nn` and `pre-nn-edges` by the `MLPEncoder`

Open DomInvivo opened this issue 2 years ago • 1 comments

Replace the pre-nn and pre-nn-edges by the MLPEncoder. Also allow pre-nn-graph and pre-nn-atten (I think this would come naturally).

In MLPEncoder, replace the use of the MLP class by the FeedForwardNN class for more options.

DomInvivo avatar Feb 27 '23 22:02 DomInvivo