We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a627a86 commit 648c512Copy full SHA for 648c512
1 file changed
sklbench/datasets/common.py
@@ -234,7 +234,7 @@ def preprocess_x(
234
scaler = MinMaxScaler(feature_range=(0, 1))
235
else:
236
logger.warning(f'Unknown "{normalize}" normalization type.')
237
- if scaler is not None and return_type == pd.DataFrame:
+ if scaler is not None:
238
return pd.DataFrame(scaler.fit_transform(x), columns=x.columns, index=x.index)
239
if return_type == np.ndarray:
240
return x.values
0 commit comments