是否可以在sklearn管道中更改熊猫列数据类型?

时间:2019-12-25 07:30:28

标签: machine-learning scikit-learn sklearn-pandas

我正在使用的Sklearn管道具有多个转换器,但其中一个初始转换器返回数字类型,而连续的转换器则使用对象类型变量。

基本上我需要挤进去:

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

管道中所需的列。

有什么办法吗?

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

1 个答案:

答案 0 :(得分:2)

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

<div id="map"></div>
<script src="https://developers.google.com/maps/documentation/javascript/examples/markerclusterer/markerclusterer.js"></script>
<!-- Replace the value of the key parameter with your own API key. -->
<script async defer src="https://maps.googleapis.com/maps/api/js?key=AIzaSyCkUOdZ5y7hMm0yrcCQoCvLwzdM6M8s5qk&callback=initMap">
</script>