Update BERTopic/bertopic/representation/_zeroshot.py
Browse files
BERTopic/bertopic/representation/_zeroshot.py
CHANGED
|
@@ -56,6 +56,7 @@ class ZeroShotClassification(BaseRepresentation):
|
|
| 56 |
topic_model,
|
| 57 |
documents: pd.DataFrame,
|
| 58 |
c_tf_idf: csr_matrix,
|
|
|
|
| 59 |
topics: Mapping[str, List[Tuple[str, float]]]
|
| 60 |
) -> Mapping[str, List[Tuple[str, float]]]:
|
| 61 |
""" Extract topics
|
|
|
|
| 56 |
topic_model,
|
| 57 |
documents: pd.DataFrame,
|
| 58 |
c_tf_idf: csr_matrix,
|
| 59 |
+
top_n_words: int,
|
| 60 |
topics: Mapping[str, List[Tuple[str, float]]]
|
| 61 |
) -> Mapping[str, List[Tuple[str, float]]]:
|
| 62 |
""" Extract topics
|