아래 링크를 통해 이 노트북을 주피터 노트북 뷰어(nbviewer.org)로 보거나 구글 코랩(colab.research.google.com)에서 실행할 수 있습니다.
주피터 노트북 뷰어로 보기 | 구글 코랩(Colab)에서 실행하기 |
# 노트북이 코랩에서 실행 중인지 체크합니다.
import os
import sys
if 'google.colab' in sys.modules:
# 사이킷런 최신 버전을 설치합니다.
!pip install -q --upgrade scikit-learn
━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 10.8/10.8 MB 20.0 MB/s eta 0:00:00
from sklearn.base import BaseEstimator, TransformerMixin
class MyTransformer(BaseEstimator, TransformerMixin):
def __init__(self, first_paramter=1, second_parameter=2):
# __init__ 메소드에 필요한 모든 매개변수를 나열합니다
self.first_paramter = 1
self.second_parameter = 2
def fit(self, X, y=None):
# fit 메소드는 X와 y 매개변수만을 갖습니다
# 비지도 학습 모델이더라도 y 매개변수를 받도록 해야합니다!
# 모델 학습 시작
print("모델 학습을 시작합니다")
# 객체 자신인 self를 반환합니다
return self
def transform(self, X):
# transform 메소드는 X 매개변수만을 받습니다
# X를 변환합니다
X_transformed = X + 1
return X_transformed