개발학습일지

[AI] 데이터 불균형 맞추기 _SMOTE() 본문

AI/데이터 전처리

[AI] 데이터 불균형 맞추기 _SMOTE()

처카푸 2024. 4. 16. 17:45

데이터 불균형 맞추기

 

필요한 라이브러리 임폴트

- 설치 안되어 있을 경우 pip install 검색 _ 구글에 pypi 검색해서 사이트에서 필요한 라이브러리 설치 명령어 찾기

from imblearn.over_sampling import SMOTE

 

SMOTE()

-  균형이 맞게 데이터 늘리거나 삭제하는 방법이 있는데,

   일반적으로 늘리는 방법을 사용한다

- 늘리는 방법으로 코드를 작성해보자

# 변수에 저장하여 사용
sm = SMOTE(random_state=2)
X, y = sm.fit_resample(X, y)