Struct Rescal 

pub struct Rescal {
    pub entity_emb: Vec<Vec<f32>>,
    pub relation_mat: Vec<Vec<Vec<f32>>>,
    pub dim: usize,
}

RESCAL: Bilinear model for knowledge graph embedding.

Each relation has a full dim×dim matrix. Scoring: h^T * M_r * t

Fields

entity_emb: Vec<Vec<f32>>

Entity embeddings: [num_entities][dim]

relation_mat: Vec<Vec<Vec<f32>>>

Relation matrices: [num_relations][dim][dim]

dim: usize

Embedding dimension

Implementations

impl Rescal

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

Create a new RESCAL model.

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

Compute score = h^T * M_r * t

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

SGD update step minimizing squared loss (score - label)^2.

pub fn relation_matrix(&self, relation: usize) -> &Vec<Vec<f32>>

Access the relation matrix for a given relation.

pub fn entity_count(&self) -> usize

Number of entities

pub fn relation_count(&self) -> usize

Number of relations

Trait Implementations

impl Clone for Rescal

fn clone(&self) -> Rescal

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Rescal

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

Formats the value using the given formatter.