Replace deprecated scikit-learn method.

EducationalTestingService · Jan 19, 2023 · 92a2e20 · 92a2e20
1 parent ffc49bd
commit 92a2e20
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rsmtool/modeler.py b/rsmtool/modeler.py
@@ -192,7 +192,7 @@ def skll_learner_params_to_dataframe(learner):
         # get the intercept, coefficients, and feature names
         intercept = learner.model.intercept_
         coefficients = learner.model.coef_
-        feature_names = learner.feat_vectorizer.get_feature_names()
+        feature_names = learner.feat_vectorizer.get_feature_names_out()
 
         # first create a sorted data frame for all the non-intercept features
         df_non_intercept = pd.DataFrame({"feature": feature_names, "coefficient": coefficients})