Struct fetish_lib::embedding_space::EmbeddingSpace

pub struct EmbeddingSpace<'a> {
    pub type_id: TypeId,
    pub model_prior_specification: &'a dyn PriorSpecification,
    pub elaborator: Elaborator<'a>,
    pub models: HashMap<TermIndex, TermModel<'a>>,
    pub ctxt: &'a Context,
}

Collection of all learned information about embeddings for a given type in and EmbedderState. This comprises the learned Elaborators and the learned TermModels for the type.

Fields

type_id: TypeId

model_prior_specification: &'a dyn PriorSpecification

PriorSpecification to use for any newly-created TermModels.

elaborator: Elaborator<'a>

models: HashMap<TermIndex, TermModel<'a>>

ctxt: &'a Context

Implementations

impl<'a> EmbeddingSpace<'a>

pub fn sample(&self, rng: &mut ThreadRng) -> SampledEmbeddingSpace<'a>

Draws a sample from the distribution defined by this EmbeddingSpace over collections of TermModels of the same type, to yield a corresponding SampledEmbeddingSpace containing information about sampled embeddings.

pub fn schmear_to_prior(
    &self,
    embedder_state: &EmbedderState<'_>,
    elaborator_func_schmear: &FuncSchmear,
    func_ptr: TermPointer,
    in_schmear: &Schmear
) -> NormalInverseWishart

pub fn add_model(&mut self, model_key: TermIndex)

Adds a new TermModel with the assigned TermIndex.

pub fn get_model_mut(&mut self, model_key: TermIndex) -> &mut TermModel<'a>

Gets a handle to the TermModel with the given TermIndex.

pub fn get_model(&self, model_key: TermIndex) -> &TermModel<'a>

Gets a reference to the TermModel with the given TermIndex.

pub fn has_model(&self, model_key: TermIndex) -> bool

Determines if a TermModel exists with the given TermIndex.

pub fn new(
    type_id: TypeId,
    model_prior_specification: &'a dyn PriorSpecification,
    elaborator_prior_specification: &dyn PriorSpecification,
    ctxt: &'a Context
) -> EmbeddingSpace<'a>

Constructs a new embedding space with the given PriorSpecifications for TermModels and for the Elaborator, respectively, and occurring within the given Context.