vectorizer.get_feature_names method update due to scikit-learn>=1.2

mesolitica · Jul 26, 2024 · 21e9d47 · 21e9d47
1 parent d166194
commit 21e9d47
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