skearn 自定义转换器 [TransformerMixin, BaseEstimator, fit_transform, fit, transform]

最新推荐文章于 2025-02-17 20:02:32 发布

原创

最新推荐文章于 2025-02-17 20:02:32 发布 · 1.1w 阅读

34 ·

CC 4.0 BY-SA版权

文章标签：

#pytohn #pandas #sklearn

自定义转换函器：
创建一个类，实现fit()[return self]、transform()和fit_transform(),
如果使用TransformerMixin作为基类，则自动实现fit_transform()函数，fit_transform() <==> fit().transform(),
如果添加BaseEstimator作为基类，,注意此时__init__函数不能接受 $* a r g s$ 和 $* * k w a r g s$ ，还可以使用两个额外的方法（get_params()和set_params()）,
这两个函数对于自动超参数调优有用处，下面是个列子，结合两个已经存在的属性作为一个新的属性值。

np.c_的用法https://blog.youkuaiyun.com/dss_dssssd/article/details/82824588

from sklearn.base import BaseEstimator, TransformerMixin
# xxx_ix 表示第几列
room_ix, bedroom_ix, population_ix, household_ix = 3, 4, 5, 6
housing[:5]

# out:
     	longitude	latitude	housing_median_age	total_rooms	total_bedrooms	population	households	median_income	ocean_proximity
17606 	-121.89	     37.29	         38.0	          1568.0	   351.0	       710.0	  339.0  	2.7042  	<1H OCEAN