GraphEncoder

oxirs_vec::cross_modal_embeddings

Trait GraphEncoder 

Source 
pub trait GraphEncoder: Send + Sync {
    // Required methods
    fn encode(&self, graph: &GraphData) -> Result<Vector>;
    fn encode_node(&self, node: &GraphNode) -> Result<Vector>;
    fn encode_subgraph(
        &self,
        nodes: &[GraphNode],
        edges: &[GraphEdge],
    ) -> Result<Vector>;
    fn get_embedding_dim(&self) -> usize;
}
Expand description

Graph encoder trait for knowledge graph embeddings

Required Methods§

Source

fn encode(&self, graph: &GraphData) -> Result<Vector>

Source

fn encode_node(&self, node: &GraphNode) -> Result<Vector>

Source

fn encode_subgraph( &self, nodes: &[GraphNode], edges: &[GraphEdge], ) -> Result<Vector>

Source

fn get_embedding_dim(&self) -> usize

Implementors§