machine-learning - 是否可以在 sklearn 管道中更改 pandas 列数据类型？

Question

我使用的 Sklearn 管道有多个变压器，但其中一个初始变压器返回数字类型，而连续的变压器则采用对象类型变量。

基本上我需要挤入:

data[col] = data[col].astype(object)

管道中所需的列。

有什么办法吗？

注意:我正在使用Feature-engine变压器。

Answer 1

是的，您可以使用 sklearn.preprocessing.FunctionTransformer 。一个简单的例子是，

def to_object(x):
  return pd.DataFrame(x).astype(object)

fun_tr = FunctionTransformer(to_object)

y = fun_tr.fit_transform(pd.DataFrame({'a':[1,2,3]}))