반응형
Notice
Recent Posts
Recent Comments
Link
관리 메뉴

bro's coding

sklearn.base.BaseEstimator, TransformerMixin(추정기 만들기) 본문

[AI]/python.sklearn

sklearn.base.BaseEstimator, TransformerMixin(추정기 만들기)

givemebro 2020. 4. 27. 11:13
반응형
from sklearn.base import BaseEstimator, TransformerMixin
class Myclassifier(BaseEstimator,TransformerMixin):
    def __init__(self,first_parameter=1,second_parameter=2):
        # __init__ 메소드에 필요한 모든 매개변수를 나열함
        self.first_parameter=1
        self.second_parameter=2
    def fit(self,X,y=None):
        # fit 메소드는 X와 y매개변수만을 갖음
        # 비지도 학습 모델이라도 y값을 받아야함
        self.result
        # 모델 학습
        print('모델 학습을 시작함!')
        return self
    
    def predict(self,X):
        # X만 받음
        X_transforms=X+1
        return X_transforms
반응형
Comments