kisejin commited on
Commit
6a57b68
·
verified ·
1 Parent(s): d36ef93

Update BERTopic/bertopic/representation/_zeroshot.py

Browse files
BERTopic/bertopic/representation/_zeroshot.py CHANGED
@@ -56,7 +56,6 @@ class ZeroShotClassification(BaseRepresentation):
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
 
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