Fixed issue where feature_names wasn't propagating to EBM after refac…

…tor.
interpretml · Dec 12, 2019 · 3aaf4fd · 3aaf4fd
1 parent fdfd531
commit 3aaf4fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/interpret-core/interpret/glassbox/ebm/ebm.py b/python/interpret-core/interpret/glassbox/ebm/ebm.py
@@ -710,7 +710,7 @@ def fit(self, X, y):  # noqa: C901
 
         # Build preprocessor
         self.preprocessor_ = EBMPreprocessor(
-            schema=self.schema, binning_strategy=self.binning_strategy, feature_types=self.feature_types
+            schema=self.schema, binning_strategy=self.binning_strategy, feature_names=self.feature_names, feature_types=self.feature_types
         )
         self.preprocessor_.fit(X)