H2GB.encoder.DistMultNodeEncoder

class DistMultNodeEncoder(dim_emb, dataset, reshape_x=True)[source]

Bases: HeteroPENodeEncoder

Knowledge Graph Embedding node encoder, based on the DistMult model from the “Embedding Entities and Relations for Learning and Inference in Knowledge Bases” paper.