Update keybert/_llm.py

Co-authored-by: Maarten Grootendorst <[email protected]>
MaartenGr · Jul 15, 2024 · 0c61bd7 · 0c61bd7
1 parent 6e76327
commit 0c61bd7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keybert/_llm.py b/keybert/_llm.py
@@ -44,7 +44,7 @@ def extract_keywords(
             docs: The document(s) for which to extract keywords/keyphrases
             check_vocab: Only return keywords that appear exactly in the documents
             candidate_keywords: Candidate keywords for each document
-            threshold: TODO
+            threshold: Minimum similarity value between 0 and 1 used to decide how similar documents need to receive the same keywords.
             embeddings: TODO
 
         Returns: