Struct KgCompletionTask 

pub struct KgCompletionTask { /* private fields */ }

Generates negative triples for knowledge graph completion training.

The task maintains an optional list of known head entities and known tail entities for type-constrained sampling; if not provided, uniform sampling is used as a fallback.

Implementations

impl KgCompletionTask

pub fn new(known_entities: Vec<String>) -> Self

Create a task with a flat list of entities (used for uniform sampling).

pub fn with_type_constraints( known_entities: Vec<String>, head_entities: Vec<String>, tail_entities: Vec<String>, ) -> Self

Create a task with separate head and tail entity pools for type-constrained sampling.

pub fn from_triples(triples: &[Triple]) -> Self

Build a KgCompletionTask by inspecting a set of observed triples.

Extracts all unique entities, heads, and tails automatically.

pub fn sample_negatives( &self, triple: &Triple, _entity_count: usize, n: usize, strategy: &NegativeSampler, ) -> Vec<Triple>

Sample n negative triples for the given positive triple.

• entity_count — not used internally (the task uses its own entity pool), but kept in the signature for API compatibility.
• strategy — sampling strategy to apply.

Returns an empty Vec when the entity pool is empty.

Trait Implementations

impl Clone for KgCompletionTask

fn clone(&self) -> KgCompletionTask

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for KgCompletionTask

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

Formats the value using the given formatter.