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.