minor fix

thisismygitrepo · Dec 7, 2023 · faf1365 · faf1365
1 parent 50afac9
commit faf1365
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/myresources/crocodile/deeplearning_df.py b/myresources/crocodile/deeplearning_df.py
@@ -201,7 +201,7 @@ def fit(self, df: 'pd.DataFrame'):
         self.clipper_categorical.fit(df=df.loc[:, self.cols_ordinal + self.cols_onehot])
         self.encoder_onehot.fit(df[self.cols_onehot])
         self.encoder_ordinal.fit(df[self.cols_ordinal])
-        self.clipper_numerical.fit(df[self.cols_numerical])
+        self.clipper_numerical.fit(df.loc[:, self.cols_numerical])
         self.imputer.fit(df[self.cols_numerical])
         self.scaler.fit(df[self.cols_ordinal + self.cols_numerical])