Struct DistMultScoring 

pub struct DistMultScoring {
    pub entity_embeds: Array2<f64>,
    pub relation_embeds: Array2<f64>,
}

DistMult bilinear scoring model (Yang et al. 2015).

Score function:

  score(h, r, t) = Σ_k  h_k · r_k · t_k

This is a symmetric scoring function: score(h,r,t) = score(t,r,h).

Fields

entity_embeds: Array2<f64>

Entity embedding table (n_entities, dim)

relation_embeds: Array2<f64>

Relation embedding table (n_relations, dim)

Implementations

impl DistMultScoring

pub fn new(n_entities: usize, n_relations: usize, dim: usize) -> Self

Create a new DistMult scorer with Xavier-initialised embeddings.

pub fn score(&self, h: usize, r: usize, t: usize) -> f64

Compute DistMult score for triple (h, r, t).

Trait Implementations

impl Clone for DistMultScoring

fn clone(&self) -> DistMultScoring

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DistMultScoring

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

Formats the value using the given formatter.

impl KgScorer for DistMultScoring

fn score(&self, h: usize, r: usize, t: usize) -> f64

Return a scalar score for the triple (h, r, t).