Renamed CategoricalToNumerical

keras-team · Oct 30, 2020 · 52ea73f · 52ea73f
1 parent 1c24d39
commit 52ea73f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/autokeras/preprocessors/common.py b/autokeras/preprocessors/common.py
@@ -98,7 +98,7 @@ def get_config(self):
         return {"lookback": self.lookback, "batch_size": self.batch_size}
 
 
-class CategoricalToNumerical(preprocessor.Preprocessor):
+class CategoricalToNumericalPreprocessor(preprocessor.Preprocessor):
     """Encode the categorical features to numerical features.
 
     # Arguments

diff --git a/tests/autokeras/preprocessors/common_test.py b/tests/autokeras/preprocessors/common_test.py
@@ -28,7 +28,7 @@ def test_time_series_input_transform():
 
 def test_categorical_to_numerical_input_transform():
     x_train = np.array([["a", "ab", 2.1], ["b", "bc", 1.0], ["a", "bc", "nan"]])
-    preprocessor = common.CategoricalToNumerical(
+    preprocessor = common.CategoricalToNumericalPreprocessor(
         column_names=["column_a", "column_b", "column_c"],
         column_types={
             "column_a": "categorical",