scvi.module.AmortizedLDAPyroModule.topic_by_feature

AmortizedLDAPyroModule.topic_by_feature(n_samples)[source]

Gets a Monte-Carlo estimate of the expectation of the topic by feature matrix.

Assumes the module has already been trained.

Parameters
n_samples : intint

Number of samples to take for the Monte-Carlo estimate of the mean.

Return type

TensorTensor

Returns

A n_topics x n_input tensor containing the topic by feature matrix.