Machine Learning/Scikit-Learn

Scikit Learn 커스텀 변환기

사이킷 런에는 원하는 기능을 구현하는 변환기를 제작할 수 있다. 이는 BaseEstimator 와 TransformerMixin을 이용하여 제작하게 된다. from sklearn.base import BaseEstimator, TransformerMixin # 열 인덱스 rooms_ix, bedrooms_ix, population_ix, households_ix = 3, 4, 5, 6 class CombinedAttributesAdder(BaseEstimator, TransformerMixin): def __init__(self, add_bedrooms_per_room=True): # *args 또는 **kargs 없음 self.add_bedrooms_per_room = add_bedrooms_per_r..