[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
반응형