Merge pull request #212 from wanadzhar913/master

`vectorizer.get_feature_names` method update due to `scikit-learn>=1.2`
mesolitica · Jul 28, 2024 · d751244 · d751244
2 parents d166194 + 21e9d47
commit d751244
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/malaya/topic_model/decomposition.py b/malaya/topic_model/decomposition.py
@@ -178,7 +178,7 @@ def fit(
             corpus[i] = cleaning(corpus[i])
 
     tf = vectorizer.fit_transform(corpus)
-    tf_features = vectorizer.get_feature_names()
+    tf_features = vectorizer.get_feature_names_out()
     compose = model(n_topics).fit(tf)
     return Topic(
         tf_features, compose, corpus, compose.transform(tf), vectorizer, tf