16
16
from .base import TransformerMixin , _fit_context , clone
17
17
from .exceptions import NotFittedError
18
18
from .preprocessing import FunctionTransformer
19
- from .utils import Bunch , _safe_indexing
19
+ from .utils import Bunch
20
20
from .utils ._estimator_html_repr import _VisualBlock
21
21
from .utils ._metadata_requests import METHODS
22
22
from .utils ._param_validation import HasMethods , Hidden
@@ -1265,7 +1265,7 @@ def make_pipeline(*steps, memory=None, verbose=False):
1265
1265
return Pipeline (_name_estimators (steps ), memory = memory , verbose = verbose )
1266
1266
1267
1267
1268
- def _transform_one (transformer , X , y , weight , columns = None , params = None ):
1268
+ def _transform_one (transformer , X , y , weight , params = None ):
1269
1269
"""Call transform and apply weight to output.
1270
1270
1271
1271
Parameters
@@ -1282,17 +1282,11 @@ def _transform_one(transformer, X, y, weight, columns=None, params=None):
1282
1282
weight : float
1283
1283
Weight to be applied to the output of the transformation.
1284
1284
1285
- columns : str, array-like of str, int, array-like of int, array-like of bool, slice
1286
- Columns to select before transforming.
1287
-
1288
1285
params : dict
1289
1286
Parameters to be passed to the transformer's ``transform`` method.
1290
1287
1291
1288
This should be of the form ``process_routing()["step_name"]``.
1292
1289
"""
1293
- if columns is not None :
1294
- X = _safe_indexing (X , columns , axis = 1 )
1295
-
1296
1290
res = transformer .transform (X , ** params .transform )
1297
1291
# if we have a weight for this transformer, multiply output
1298
1292
if weight is None :
@@ -1301,14 +1295,7 @@ def _transform_one(transformer, X, y, weight, columns=None, params=None):
1301
1295
1302
1296
1303
1297
def _fit_transform_one (
1304
- transformer ,
1305
- X ,
1306
- y ,
1307
- weight ,
1308
- columns = None ,
1309
- message_clsname = "" ,
1310
- message = None ,
1311
- params = None ,
1298
+ transformer , X , y , weight , message_clsname = "" , message = None , params = None
1312
1299
):
1313
1300
"""
1314
1301
Fits ``transformer`` to ``X`` and ``y``. The transformed result is returned
@@ -1317,9 +1304,6 @@ def _fit_transform_one(
1317
1304
1318
1305
``params`` needs to be of the form ``process_routing()["step_name"]``.
1319
1306
"""
1320
- if columns is not None :
1321
- X = _safe_indexing (X , columns , axis = 1 )
1322
-
1323
1307
params = params or {}
1324
1308
with _print_elapsed_time (message_clsname , message ):
1325
1309
if hasattr (transformer , "fit_transform" ):
0 commit comments