Skip to main content

train

qualia_core_db::solvers::learning::kg_embedding::train

Function train 

Source 
pub fn train(
    triples: &[(usize, usize, usize)],
    n_entities: usize,
    n_relations: usize,
    cfg: TrainConfig,
) -> Result<EmbeddingTable, KgEmbeddingError>
Expand description

Train an embedding table on triples (entity/relation indices) with n_entities distinct entities and n_relations relations. Returns the trained table, or fails closed on an empty corpus / inconsistent config.