H2GB.encoder.RawNodeEncoder

class RawNodeEncoder(dim_emb, dataset)[source]

Bases: Module

The raw feature node encoder.

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