H2GB.encoder.HeteroRawNodeEncoder

class HeteroRawNodeEncoder(dim_emb, dataset)[source]

Bases: Module

The heterogeneous raw feature node encoder.

Apply a type-specific linear transformation to each node feature to transform them into a unified node embedding sapce.

Parameters:

  • emb_dim (int) – Output embedding dimension

  • dataset (Any) – A InMemoryDataset dataset object.