Index

B

BaseBowEstimator (class in tmnt.estimator) BaseDistribution (class in tmnt.distribution) BaseEstimator (class in tmnt.estimator) BaseInferencer (class in tmnt.inference) BaseSeqBowVED (class in tmnt.modeling)

BaseVAE (class in tmnt.modeling) BowEstimator (class in tmnt.estimator) BowMetricEstimator (class in tmnt.estimator) BowVAEInferencer (class in tmnt.inference) BowVAEModel (class in tmnt.modeling)

C

CoherenceRegularizer (class in tmnt.modeling) ContinuousCovariateModel (class in tmnt.modeling)

CovariateBowEstimator (class in tmnt.estimator) CovariateBowVAEModel (class in tmnt.modeling) CovariateModel (class in tmnt.modeling)

E

encode_data() (BowVAEModel method)

encode_data_with_covariates() (CovariateBowVAEModel method)

F

fit() (BaseBowEstimator method) (BaseEstimator method) fit_transform() (TMNTVectorizer method) fit_transform_json() (TMNTVectorizer method) fit_transform_json_dir() (TMNTVectorizer method) fit_with_validation() (BaseBowEstimator method) (BaseEstimator method) (SeqBowEstimator method) forward() (BowVAEModel method) (CoherenceRegularizer method) (ContinuousCovariateModel method) (CovariateBowVAEModel method) (CovariateModel method) (GaussianDistribution method) (GaussianUnitVarDistribution method) (GeneralizedSDMLLoss method) (LogisticGaussianDistribution method) (MetricBowVAEModel method) (MetricSeqBowVED method) (MultiNegativeCrossEntropyLoss method) (SeqBowVED method) (VonMisesDistribution method)

from_config() (BaseBowEstimator class method) (BowEstimator class method) (BowMetricEstimator class method) (CovariateBowEstimator class method) (SeqBowEstimator class method) (SeqBowMetricEstimator class method) from_saved() (BaseBowEstimator class method) (BowEstimator class method) from_vocab_file() (TMNTVectorizer class method)

G

GaussianDistribution (class in tmnt.distribution) GaussianUnitVarDistribution (class in tmnt.distribution) GeneralizedSDMLLoss (class in tmnt.modeling) get_likelihood_stats() (BowVAEInferencer method) get_llm() (in module tmnt.data_loading) get_llm_dataloader() (in module tmnt.data_loading) get_llm_model() (in module tmnt.data_loading) get_llm_paired_dataloader() (in module tmnt.data_loading) get_llm_tokenizer() (in module tmnt.data_loading) get_mu_encoding() (GaussianDistribution method) (GaussianUnitVarDistribution method) (LogisticGaussianDistribution method) (VonMisesDistribution method)

get_ordered_terms() (BaseSeqBowVED method) (BaseVAE method) get_ordered_terms_with_covar_at_data() (CovariateBowVAEModel method) get_topic_vectors() (BaseVAE method) (BowEstimator method) (CovariateBowEstimator method) (CovariateBowVAEModel method) get_unwrapped_llm_dataloader() (in module tmnt.data_loading) get_vocab() (TMNTVectorizer method)

L

load_vocab() (in module tmnt.data_loading), [1]

log_train() (SeqBowEstimator method) LogisticGaussianDistribution (class in tmnt.distribution)

M

MetricBowVAEModel (class in tmnt.modeling) MetricSeqBowVED (class in tmnt.modeling) MetricSeqVEDInferencer (class in tmnt.inference) module tmnt.data_loading tmnt.distribution tmnt.estimator tmnt.inference tmnt.modeling tmnt.preprocess.vectorizer

MultiNegativeCrossEntropyLoss (class in tmnt.modeling)

P

perplexity() (BowEstimator method)

predict() (BowVAEModel method) predict_text() (BowVAEInferencer method)

S

save() (BaseInferencer method) SeqBowEstimator (class in tmnt.estimator) SeqBowMetricEstimator (class in tmnt.estimator) SeqBowVED (class in tmnt.modeling) SeqVEDInferencer (class in tmnt.inference)

sparse_batch_collate() (in module tmnt.data_loading), [1] sparse_coo_to_tensor() (in module tmnt.data_loading), [1] SparseDataLoader (class in tmnt.data_loading) SparseDataset (class in tmnt.data_loading) StratifiedDualBatchSampler (class in tmnt.data_loading)

T

tmnt.data_loading module tmnt.distribution module tmnt.estimator module tmnt.inference module tmnt.modeling module

tmnt.preprocess.vectorizer module TMNTVectorizer (class in tmnt.preprocess.vectorizer) to_label_matrix() (in module tmnt.data_loading), [1] transform() (BowEstimator method) (CovariateBowEstimator method) (TMNTVectorizer method) transform_json() (TMNTVectorizer method) transform_json_dir() (TMNTVectorizer method)

V

VonMisesDistribution (class in tmnt.distribution)

W

write_model() (SeqBowEstimator method)

write_to_vec_file() (TMNTVectorizer method) write_vocab() (TMNTVectorizer method)