Module models 

Embedding model implementations

This module provides various knowledge graph embedding models including:

• TransE: Translation-based embeddings
• ComplEx: Complex number embeddings for asymmetric relations
• DistMult: Bilinear diagonal model
• RotatE: Rotation-based embeddings
• HolE: Holographic embeddings using circular correlation
• ConvE: Convolutional embeddings with 2D CNNs (optional)
• TuckER: Tucker decomposition based embeddings (optional)
• TransformerEmbedding: Transformer-based embeddings (BERT, RoBERTa, etc.)
• GNNEmbedding: Graph Neural Network embeddings (GCN, GraphSAGE, GAT, etc.)
• OntologyAwareEmbedding: Embeddings that respect RDF/OWL ontology constraints

Re-exports

pub use complex::ComplEx;

pub use distmult::DistMult;

pub use gnn::AggregationType;

pub use gnn::GNNConfig;

pub use gnn::GNNEmbedding;

pub use gnn::GNNType;

pub use hole::HoLE;

pub use hole::HoLEConfig;

pub use ontology::OntologyAwareConfig;

pub use ontology::OntologyAwareEmbedding;

pub use ontology::OntologyConstraints;

pub use ontology::OntologyRelation;

pub use rotate::RotatE;

pub use transe::TransE;

pub use transformer::PoolingStrategy;

pub use transformer::TransformerConfig;

pub use transformer::TransformerEmbedding;

pub use transformer::TransformerType;

pub use conve::ConvE;

pub use conve::ConvEConfig;

pub use tucker::TuckER;

pub use quatd::QuatD;

pub use scirs_neural::ActivationType;

pub use scirs_neural::OptimizerType;

pub use scirs_neural::SciRS2NeuralConfig;

pub use scirs_neural::SciRS2NeuralEmbedding;

pub use base::*;

pub use common::*;

Modules

base

Base functionality shared across embedding models

common

Common utilities and functions used across embedding models

complex

ComplEx: Complex Embeddings for Simple Link Prediction

conve

ConvE (Convolutional Embeddings) Model

distmult

DistMult: Embedding Entities and Relations for Learning and Inference in Knowledge Bases

gnn

Graph Neural Network (GNN) embedding models

hole

HolE (Holographic Embeddings) Model

ontology

Ontology-aware embedding models

quatd

QuatE: Quaternion Embeddings for Knowledge Graph Completion

rotate

RotatE: Rotation-based Knowledge Graph Embeddings

scirs_neural

SciRS2 Neural Network Integration for Enhanced Embeddings

transe

TransE: Translating Embeddings for Modeling Multi-relational Data

transformer

Transformer-based embedding models module

tucker

TuckER: Tucker Decomposition for Knowledge Graph Embeddings