使用sklearn.compose.ColumnTransformer进行批量数据转换

sklearn.compose.ColumnTransformer可以用来构建一个数据转换器,用允许单独转换输入的不同列或列子集,并且每个转换器生成的特征将串联起来以形成单个特征空间。这对于异构或列式数据非常有用,可以将多个特征提取机制或转换组合到单个转换器中。
配合sklearn.pipeline.make_pipeline可以高效地构建一个可以复用的数据转换器,常用的可以配合使用StandardScaler以及Normalizer

from sklearn.pipeline import make_pipeline
from sklearn.compose import make_column_transformer
from sklearn.preprocessing import StandardScaler, MinMaxScaler

X = train.drop('target', axis=1).copy()
y = train.target.copy()
test_X = test.copy()

# Scaling and Nomalization
transformer = make_pipeline(
    StandardScaler()
)
columns = X.columns[:-11]

transformer_new = make_pipeline(
    StandardScaler()
)
new_columns = X.columns[-11:]

preprocessor = make_column_transformer(
    (transformer, columns),
    (transformer_new, new_columns),
)

使用:

train_X = preprocessor.fit_transform(train_X)
val_X = preprocessor.transform(val_X)
posted @   Asp1rant  阅读(466)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· AI技术革命,工作效率10个最佳AI工具
历史上的今天:
2019-05-07 两种语言实现设计模式(C++和Java)(七:原型模式)
点击右上角即可分享
微信分享提示