Where are the docs, TopDocs?

[Alvant]

What is the matter

Seems like TopDocuments Viewer assigns each document to one topic only. Even if there are some other topics which are represented well in the document. (These topics won't have the document in the result view.)

How to reproduce

1. Make a small dataset of, let's say, three documents.
2. Create a topic model of, let's say, 50 topics. Fit it on the dataset.
3. Create a TopDocumentsViewer. Get a view of the model's topics' documents.

Result

Some topics do not have any documents, even if the probabilities in the Theta matrix are high.

where the view_model function is:

Expected result

There is a way to control which documents are considered as "top documents". For example, if several topics have high probabilities in a particular document, then maybe there should be an opportunity to put this document in the "top lists of documents" for all the aforementioned topics.

[Alvant]

Seems like predict_cluster_by_precomputed_distances is responsible for this behaviour.