relbench icon indicating copy to clipboard operation
relbench copied to clipboard

Question: Learning Embedding vectors

Open pmavrodiev opened this issue 1 year ago • 1 comments
trafficstars

Section 3.4.1 of the paper describes Heterogeneous Message Passing as a generic framework to learn embedding vectors ${{h_{v}^{(L)}}}_{\nu \in V}$ for each node, which are then fed to the model heads.

Do you intend to offer an implementation of the heterogeneous message passing in this package?

pmavrodiev avatar Jan 05 '24 16:01 pmavrodiev

Yes! RelBench is currently in beta version. We will provide baseline implementations in a few months. Stay tuned

kexinhuang12345 avatar Jan 14 '24 05:01 kexinhuang12345

Please see examples directory for the GNN training scripts.

rishabh-ranjan avatar Jul 22 '24 21:07 rishabh-ranjan