Struct DynamicTopicModel 

pub struct DynamicTopicModel {
    pub config: DtmConfig,
    /* private fields */
}

Dynamic Topic Model estimator.

Fit via DynamicTopicModel::fit; the result is returned as a DtmResult.

Fields

config: DtmConfig

Model configuration.

Implementations

impl DynamicTopicModel

pub fn fit( &self, docs_by_time: &[Vec<Vec<f64>>], vocab_size: usize, ) -> Result<DtmResult>

Fit the DTM to a corpus organised by time slice.

Arguments

• docs_by_time – T lists of documents; each document is a word-count vector of length V
• vocab_size – vocabulary size V

Returns

A DtmResult with topic-word trajectories and doc-topic distributions.

impl DynamicTopicModel

pub fn new(config: DtmConfig) -> Self

Construct a new (unfitted) DTM with the given configuration.

pub fn fitted_result(&self) -> Option<&DtmResult>

Return a reference to the fitted result, if available.

pub fn fit_and_store( &mut self, docs_by_time: &[Vec<Vec<f64>>], vocab_size: usize, ) -> Result<&DtmResult>

Fit the model and store the result internally, also returning it.

pub fn top_words_at( &self, t: usize, vocab: &[String], n: usize, ) -> Option<Vec<Vec<String>>>

Return the top-n words for each topic at time t using the fitted model.

pub fn word_evolution( &self, topic_id: usize, word_id: usize, ) -> Option<Vec<f64>>

Return the evolution of word word_id in topic topic_id over time.

Trait Implementations

impl Default for DynamicTopicModel

fn default() -> Self

Returns the “default value” for a type.