H2GB.encoder.HeteroRawEdgeEncoder

class HeteroRawEdgeEncoder(dim_emb, dataset)[source]

Bases: Module

The heterogeneous raw feature edge encoder.

Apply a type-specific 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.