scikit-learn中FunctionTransformer的feature_names_out重命名失效如何修复

你的代码问题出在feature_names_out参数的自定义函数签名不符合要求——sklearn要求这个函数必须接受两个参数：第一个是Transformer实例（可以用下划线忽略），第二个是输入的特征名称列表。你之前只写了一个参数，导致该函数没有被正确触发，所以列名没变化。

修复后的代码如下：

import pandas as pd
from sklearn.preprocessing import FunctionTransformer
from sklearn import set_config

set_config(transform_output='pandas') # 确保输出为DataFrame

df = pd.DataFrame(
    {
        'x1' : [1, 2, 3],
        'x2' : [2, 3, 4]
    }
)

# 修改feature_names_out的lambda，接受两个参数，用_忽略第一个无关参数
my_transformer = FunctionTransformer(lambda x: x**2,
                    feature_names_out= lambda _, features : [f's2_{c}' for c in features])

transformed_df = my_transformer.fit_transform(df)
transformed_df

运行后transformed_df的列名将变为s2_x1和s2_x2，符合你的预期。

内容的提问来源于stack exchange，提问作者PingPong