Merge pull request #767 from a-szulc/master

add setting `feature_name_in` parameter in `scale` class (fixes #759 and #760)
mljar · Aug 29, 2024 · 19272cd · 19272cd
2 parents c3f9ac8 + 3300a55
commit 19272cd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/supervised/preprocessing/scale.py b/supervised/preprocessing/scale.py
@@ -83,6 +83,7 @@ def from_json(self, data_json):
         self.scale.n_samples_seen_ = int(data_json.get("n_samples_seen"))
         self.scale.n_features_in_ = int(data_json.get("n_features_in"))
         self.columns = data_json.get("columns", [])
+        self.scale.feature_names_in_ = data_json.get("columns")
         self.scale_method = data_json.get("scale_method")
         self.X_min_values = data_json.get("X_min_values")
         if self.X_min_values is not None: