H2GB.encoder.RawEdgeEncoder

class RawEdgeEncoder(dim_emb, dataset)[source]

Bases: Module

The raw feature edge encoder.

Apply a linear transformation to each edge feature to transform them into a unified edge embedding sapce.

Parameters:

  • emb_dim (int) – Output embedding dimension

  • dataset (Any) – A InMemoryDataset dataset object.