Struct RotatEPlus 

pub struct RotatEPlus {
    pub entity_phase: Vec<Vec<f32>>,
    pub relation_phase: Vec<Vec<f32>>,
    pub dim: usize,
}

RotatE+ with phase-space operations.

Entities and relations are represented as phase vectors in [0, 2π). The scoring function computes the L1 distance in phase space after applying the relational rotation: score = -||h ∘ r - t||_1.

Fields

entity_phase: Vec<Vec<f32>>

Entity phase embeddings: [num_entities][dim], values in [0, 2π)

relation_phase: Vec<Vec<f32>>

Relation phase embeddings: [num_relations][dim], values in [0, 2π)

dim: usize

Embedding dimension

Implementations

impl RotatEPlus

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

Create a new RotatE+ model with random phase initialization.

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

Compute score = -||h ∘ r - t||_1 where ∘ is element-wise phase addition mod 2π.

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

Update embeddings via margin-based gradient step.

pos_score and neg_score are output from score(). We push pos_score higher and neg_score lower.

pub fn entity_count(&self) -> usize

Number of entities

pub fn relation_count(&self) -> usize

Number of relations

Trait Implementations

impl Clone for RotatEPlus

fn clone(&self) -> RotatEPlus

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for RotatEPlus

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

Formats the value using the given formatter.