From c92b495ba593715a13e9895ea0554d5127997af9 Mon Sep 17 00:00:00 2001 From: Jonas Landsgesell Date: Wed, 3 Jan 2024 09:32:38 +0100 Subject: [PATCH] Update _bertopic.py to fix question/ github issue #1696 --- bertopic/_bertopic.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/bertopic/_bertopic.py b/bertopic/_bertopic.py index 22b6c03c..d29505ce 100644 --- a/bertopic/_bertopic.py +++ b/bertopic/_bertopic.py @@ -456,7 +456,10 @@ def transform(self, """ After having fit a model, use transform to predict new instances Arguments: - documents: A single document or a list of documents to predict on + documents: A single document or a list of documents to predict on. + Note that the behavior of the method might differ depending + on whether a single document or a list of documents is passed + (especially when using the HDBSCAN algorithm) embeddings: Pre-trained document embeddings. These can be used instead of the sentence-transformer model. images: A list of paths to the images to predict on or the images themselves