Struct PairRE 

pub struct PairRE {
    pub entity_emb: Vec<Vec<f32>>,
    pub relation_head: Vec<Vec<f32>>,
    pub relation_tail: Vec<Vec<f32>>,
    pub dim: usize,
}

PairRE: Handles 1-N, N-1, and N-N relations with paired relation vectors.

Each relation has two vectors r_h (applied to head) and r_t (applied to tail). score = -||h ⊙ r_h - t ⊙ r_t||_2

Fields

entity_emb: Vec<Vec<f32>>

Entity embeddings: [num_entities][dim]

relation_head: Vec<Vec<f32>>

Head-side relation vectors: [num_relations][dim]

relation_tail: Vec<Vec<f32>>

Tail-side relation vectors: [num_relations][dim]

dim: usize

Embedding dimension

Implementations

impl PairRE

pub fn new(num_entities: usize, num_relations: usize, dim: usize) -> Self

Create a new PairRE model with small random initializations.

pub fn score(&self, head: usize, relation: usize, tail: usize) -> f32

Compute score = -||h ⊙ rh - t ⊙ rt||_2

pub fn update( &mut self, head: usize, relation: usize, tail: usize, label: f32, lr: f32, )

SGD-style update step.

pub fn predict_tail( &self, head: usize, relation: usize, top_k: usize, ) -> Vec<(usize, f32)>

Predict top-k tail entities for a (head, relation) query.

pub fn entity_count(&self) -> usize

Number of entities

Trait Implementations

impl Clone for PairRE

fn clone(&self) -> PairRE

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for PairRE

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.