앙상블 기법을 통해 여러 모델을 결합하고, 손실 함수를 활용해 예측 오류를 측정하며, 과적합을 방지하고 하이퍼파라미터 튜닝을 통해 모델 성능을 최적화하는 방법
1. 앙상블 기법
- 여러 개의 모델을 조합하여, 하나의 모델보다 더 좋은 예측 성능을 내는 방법
- why?
- 서로 다른 관점(모델)을 결합함으로써 오류를 줄일 수 있음
- 개별 모델의 편향과 분산을 상호 보완
- 배깅 (Bagging, Bootstrap Aggregating)
- 원리
- 학습 데이터를 무작위로 여러 부분 샘플(부트스트랩)로 나누어 각각 독립적으로 모델 학습
- 예측 시에는 여러 모델의 결과를 평균(회귀) 혹은 다수결(분류)로 결정
- 예시
- 랜덤포레스트 - 분류, 회귀 모두 가능
- 결정 트리 여러 개를 만들 때, 각 트리에 사용하는 피처와 데이터 샘플을 무작위로 선택 (피처 샘플링 + 데이터 샘플링)
- 결정 트리 : 데이터를 여러 조건(규칙)으로 분할하여 트리 형태로 예측을 수행하는 머신러닝 모델
- ex. 날씨 데이터를 보고 테니스를 치는게 가능한지 아닌지 예측하는 결정트리 모델 만들기
- 결정 트리 여러 개를 만들 때, 각 트리에 사용하는 피처와 데이터 샘플을 무작위로 선택 (피처 샘플링 + 데이터 샘플링)
- 랜덤포레스트 - 분류, 회귀 모두 가능
- 장점
- 각 모델이 독립적으로 학습되므로 병렬 처리 가능 (학습 속도가 상대적으로 빠름)
- 모델 간 상호 간섭이 적어 안정적
- 과적합을 줄여주는 효과 (예측의 분산 감소)
- 단점
- 많은 수의 모델을 학습해야 하므로 메모리 사용량이 많아질 수 있음
- 해석 어려움
- 코드 예시
- 원리
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import train_test_split
from sklearn.ensemble import RandomForestClassifier
from sklearn.metrics import accuracy_score, confusion_matrix, classification_report
# 1. 데이터 로드
data = load_breast_cancer()
X = data.data
y = data.target
# 2. 학습 / 데이터 분할
X_train, X_test, y_train, y_test = train_test_split(X, y,
test_size = 0.2,
random_state = 42,
stratify = y)
# 3. 랜덤 포레스트 모델 생성
# n_estimators는 사용할 트리의 개수, max_depth는 각 트리의 최대 깊이를 의미하며
# 위 2개의 값을 높일 수록 시간과 연산량은 늘어나지만 더욱 복잡한 특징을 잡을 수 있음
rf_model = RandomForestClassifier(n_estimators = 100,
max_depth = None,
random_state = 42)
# 4. 모델 학습
rf_model.fit(X_train, y_train)
# 5. 예측
y_pred = rf_model.predict(X_test)
# 6. 성능 평가
acc = accuracy_score(y_test, y_pred)
cm = confusion_matrix(y_test, y_pred)
report = classification_report(y_test, y_pred)
print(f"Accuracy: {acc:.4f}")
print(f"Confusion Matrix:\n", cm)
print("Classification Report:\n", report)
- load_breast_cancer()를 통해 이진분류용 유방암 진단 데이터셋을 불러온다. 데이터(X)와 라벨(y)을 분리합니다.
- train_test_split()으로 학습(train) 세트와 테스트(test) 세트를 8:2 비율로 분할한다. random_state로 난수를 고정하고, 클래스 비율이 유지되도록 stratify=y를 지정합니다.
- 랜덤 포레스트(RandomForestClassifier) 모델을 생성한다. n_estimators는 사용할 트리의 개수, max_depth는 각 트리의 최대 깊이를 뜻합니다.
- fit(X_train, y_train)으로 학습 데이터를 사용해 모델을 학습합니다.
- 학습된 모델로 테스트 세트(X_test)를 예측하여 y_pred를 얻습니다.
- 정확도(Accuracy), 혼동 행렬(Confusion Matrix), 분류 보고서(Classification Report)를 출력해 성능을 평가합니다.
- accuracy_score()로 정확도를 계산합니다.
- confusion_matrix()를 통해 실제 라벨 대비 예측 라벨의 분포를 살펴볼 수 있습니다.
- classification_report()로 정밀도(precision), 재현율(recall), F1 점수 등 세부 지표를 확인합니다.
- 부스팅 (Boosting)
- 원리
- 순차적으로 모델을 학습하면서 이전 모델이 만든 예측 오류를 보정하도록 설계
- 각각의 모델은 이전 모델이 틀린 부분에 가중치를 더 둬서 학습
- 대표 알고리즘 - 분류 회귀 모두 가능
- XGBoost (Extrme Gradient Boosing)
- LightGBM
- CatBoost
- 장점
- 높은 정확도 달성 가능
- 각 단계에서의 오류를 보정하기 때문에 복잡한 데이터 패턴을 잘 포착
- 단점
- 순차적으로 학습하므로 병렬화가 쉽지 않음
- 하이퍼파라미터가 많고 튜닝이 까다롭다
- 작동 예시 (XGBoost) 간단 시나리오
- 1) 기본 모델 (약한 결정 트리) 훈련 → 예측 오류 확인
- 2) 예측 오류가 컸던 샘플에 높은 가중치 부여
- 3) 다음 모델 (결정 트리) 훈련 → 다시 오류 보정
- 4) 이 과정을 여러번 반복하여, 최종 예측 시에는 모두 합산
- XGBoost, LightGBM, CatBoost코드예시
- 원리
# 1. 데이터 준비 (Titanic 예시: 범주형 컬럼 존재)
from sklearn.datasets import fetch_openml
import pandas as pd
import numpy as np
# OpenML에서 Titanic 데이터셋 로드
titanic = fetch_openml('titanic', version=1, as_frame=True)
df = titanic.frame
df
# 주요 컬럼만 사용하고, 결측치가 있는 행 제거(XGB와 Light GBM을 위해)
# pclass(객실 등급, 범주형), sex(성별, 범주형), age(나이, 연속형), fare(티켓 요금, 연속형)
# embarked(탑승항구, 범주형), survived(생존 여부, 타깃)
df = df[['pclass', 'sex', 'age', 'fare', 'embarked', 'survived']]
df.dropna(inplace=True)
df
# 입력(X), 타깃(y) 분리
X = df.drop('survived', axis=1)
y = df['survived'].astype(int) # survived 컬럼을 int형으로 변환
print(X.shape) # (1043, 5)
print(y.shape) # (1043,)
# 2. 데이터 전처리
# XGBoost/LightGBM은 숫자형 입력만 허용하므로, 범주형 칼럼을 인코딩
from sklearn.preprocessing import LabelEncoder
cat_cols = ['sex', 'embarked'] # 범주형으로 간주할 컬럼들
for col in cat_cols:
le = LabelEncoder()
X[col] = le.fit_transform(X[col])
X
# 3. 학습/테스트 데이터 분할
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(
X, y,
test_size=0.2,
random_state=42,
stratify=y
)
print(X_train.shape) # (834, 5)
print(y_train.shape) # (834,)
print(X_test.shape) # (209, 5)
print(y_test.shape) # (209,)
# 4. XGBoost 실습
# (설치가 필요할 수 있습니다) ! pip install xgboost
from xgboost import XGBClassifier
from sklearn.metrics import accuracy_score, confusion_matrix, classification_report
xgb_model = XGBClassifier(random_state=42)
xgb_model.fit(X_train, y_train)
y_pred_xgb = xgb_model.predict(X_test)
print("=== XGBoost ===")
print("Accuracy:", accuracy_score(y_test, y_pred_xgb))
print("Confusion Matrix:\n", confusion_matrix(y_test, y_pred_xgb))
print("Classification Report:\n", classification_report(y_test, y_pred_xgb))
# 5. LightGBM 실습
# (설치가 필요할 수 있습니다) ! pip install lightgbm
from lightgbm import LGBMClassifier
lgb_model = LGBMClassifier(random_state=42)
lgb_model.fit(X_train, y_train)
y_pred_lgb = lgb_model.predict(X_test)
print("\n=== LightGBM ===")
print("Accuracy:", accuracy_score(y_test, y_pred_lgb))
print("Confusion Matrix:\n", confusion_matrix(y_test, y_pred_lgb))
print("Classification Report:\n", classification_report(y_test, y_pred_lgb))
![]() |
![]() |
# 설치 필수!
! pip install catboost
# 6. CatBoost 실습 (범주형 특성 직접 지정 예시)
# -> 별도의 인코딩 없이도 'cat_features' 인덱스를 지정 가능
# 이전 예시에선 이미 LabelEncoder로 숫자로 바꿨지만,
# CatBoost는 원본 범주형(문자열) 상태로도 학습 가능.
from catboost import CatBoostClassifier
# CatBoost용 데이터 준비: 원본 df에서 결측 제거(위에서 한 것 동일)
df_cat = titanic.frame[['pclass', 'sex', 'age', 'fare', 'embarked', 'survived']].dropna()
X_cat = df_cat.drop('survived', axis=1)
y_cat = df_cat['survived'].astype(int)
X_cat
# cat_features 인덱스: 'sex', 'embarked' 컬럼(원본 df에서의 컬럼 인덱스)
# DataFrame 사용 시에는 컬럼 이름이 아니라 "열의 위치"를 지정해야 함
# - pclass : 0, sex: 1, age: 2, fare: 3, embarked: 4
cat_features_idx = [1, 4]
X_cat_train, X_cat_test, y_cat_train, y_cat_test = train_test_split(
X_cat, y_cat, test_size=0.2, random_state=42, stratify=y_cat
)
cat_model = CatBoostClassifier(
cat_features=cat_features_idx,
verbose=1, # 학습과정 확인 가능
random_state=42
)
cat_model.fit(X_cat_train, y_cat_train)
y_pred_cat = cat_model.predict(X_cat_test)
print("\n=== CatBoost ===")
print("Accuracy:", accuracy_score(y_cat_test, y_pred_cat))
print("Confusion Matrix:\n", confusion_matrix(y_cat_test, y_pred_cat))
print("Classification Report:\n", classification_report(y_cat_test, y_pred_cat))
- Titanic 데이터 로드
- fetch_openml('titanic', version=1, as_frame=True)를 사용하여 Titanic 데이터를 가져옵니다.
- 얻어온 DataFrame에서 원하는 컬럼만 골라(pclass, sex, age, fare, embarked, survived) 사용하고, 결측치가 있는 행은 제거합니다.
- 범주형 인코딩
- XGBoost나 LightGBM은 입력을 숫자형으로만 처리하므로, LabelEncoder를 사용해 sex, embarked 등 범주형 컬럼들을 수치화합니다.
- CatBoost는 별도의 인코딩 없이 cat_features 파라미터를 통해 범주형 컬럼을 처리할 수 있으나, 비교를 위해 이 과정을 진행합니다.
- 학습/테스트 데이터 분할
- train_test_split으로 80%를 학습용, 20%를 테스트용으로 사용합니다.
- stratify=y를 통해 생존 여부의 클래스 분포가 분할 시 유지되도록 합니다.
- XGBoost 실습
- XGBClassifier를 생성 후 fit으로 학습합니다.
- predict 결과와 실제값을 비교해 정확도, 혼동행렬, 분류 보고서를 출력합니다.
- LightGBM 실습
- LGBMClassifier를 생성 후 동일한 절차로 학습/예측/평가합니다.
- CatBoost 실습
- CatBoost는 범주형 데이터를 직접 지정(cat_features) 가능합니다.
- 이를 위해 원본 DataFrame(문자열 범주형)을 다시 준비하고, cat_features 인덱스를 설정합니다.
- CatBoostClassifier로 학습 후, 예측 결과를 평가합니다.
2. 과적합 (Overfitting) VS 과소적합 (Underfitting)
과적합 | 과소적합 |
![]() |
![]() |
- 과적합
- 학습 데이터에는 지나치게 최적화되었지만, 새로운 데이터(테스트/실제 환경)에는 성능이 떨어지는 현상
- 과소적합
- 모델이 데이터의 패턴을 충분히 학습하지 못해, 학습 데이터조차도 충분히 맞추지 못하는 현상
- 과적합의 원인
- 모델의 파라미터(자유도)가 너무 많아서 복잡도 과다
- 학습 데이터 수가 충분하지 않음
- 너무 많은 에폭 (딥러닝 등)으로 학습
- 노이즈가 많은 훈련 데이터에서 패턴을 '과하게' 학습
- 과적합 해결 방법
- 정규화(Regularization) 기법
- ex. L1, L2 정규화 : 가중치(모델 파라미터)에 패널티를 줘서 과도한 학습 억제
- 드롭아웃 (Dropout, 딥러닝에 주로 사용)
- 학습 시 일부 뉴런을 확률적으로 비활성화 → 과적합 완화
- 데이터 증강
- 이미지 데이터의 경우, 회전/이동/반전 등으로 새 데이터를 생성
- 자연어 데이터에도 유사한 패턴으로 증강 가능
- 신호 데이터의 경우 가우시안 노이즈를 추가하여 증강 가능
- 조기 종료
- 학습 도중 검증 손실이 증가하기 시작하면 학습 중단
- 앙상블
- 서로 다른 모델을 결합하여 과적합 위험을 줄임
- 정규화(Regularization) 기법
- 과소적합 해결 방법
- 모델 복잡도 증가
- 더 오래 학습
- 모델 구조 변경 (더 깊은 신경망, 더 많은 트리 등)
3. 하이퍼파라미터 튜닝
- 모델이 학습을 시작하기 전에 사람이 설정해야 하는 값
- ex. 결정 트리의 최대 깊이, 학습 횟수 등
- 하이퍼파라미터 튜팅을 위한 데이터 준비
- 데이터셋 분할 (Training/Validation/Test)
- Training Set: 모델 학습에 직접 사용
- Validation Set: 하이퍼파라미터 튜닝이나 모델 선택을 위해 사용
- Test Set: 최종 성능 평가(훈련/검증 단계에 절대 포함되면 안 됨)
- 교차 검증 (Cross - Validation)
- 데이터를 훈련 세트와 검증 세트로 여러 번 겹치지 않게 나누어 사용
- K-Fold Cross-Validation:
- 데이터를 K개의 폴드(Fold)로 나누어, 순차적으로 한 폴드를 검증 세트로 사용하고 나머지를 훈련에 사용
- 평균 성능을 최종 모델의 성능으로 본다
- 장점 : 데이터가 적은 상황에서도 안정적인 성능 평가 가능
- 데이터셋 분할 (Training/Validation/Test)
- 튜닝 방법
- 1. Grid Search
- 미리 정의된 하이퍼파라미터 후보들의 '모든 조합'을 시도
- 장점 : 완전 탐색이므로 최적값을 놓치지 않음
- 단점 : 후보가 많아질수록 연산량이 급격히 증가
- 2. Randomized Search
- 임의로 샘플링된 하이퍼파라미터 조합을 일정 횟수만 시도
- 장점 : 다양한 영역을 빠르게 탐색 가능, 속도 빠름
- 단점 : 최적 조합을 정확히 찾지 못할 수도 있음
- 3. 베이지안 최적화
- 과거의 탐색 결과를 바탕으로 '가장 유망한 하이퍼파라미터 범위'를 중점적으로 탐색
- 장점 : 탐색 시간이 더 짧고 효율적
- 단점 : 구현 복잡도가 높음
- 1번 Grid SearchCV 코드 예시
- 1. Grid Search
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split, GridSearchCV
from sklearn.ensemble import RandomForestClassifier
from sklearn.metrics import accuracy_score
# 1. 데이터 로드
iris = load_iris()
X = iris.data
y = iris.target
# 2. 학습/테스트 데이터 분할
X_train, X_test, y_train, y_test = train_test_split(
X, y,
test_size=0.2,
random_state=42,
stratify=y
)
# 3. 하이퍼 파라미터 후보군 설정
param_grid = {
'n_estimators': [50, 100, 200],
'max_depth': [None, 5, 10]
}
# 4. GridSearchCV 생성
rf = RandomForestClassifier(random_state=42)
grid_search = GridSearchCV(
estimator=rf,
param_grid=param_grid,
cv=5, # 교차검증(fold) 횟수
scoring='accuracy',
n_jobs=-1, # 병렬 처리(가능한 모든 코어 사용)
)
# 5. 학습(그리드서치 수행)
grid_search.fit(X_train, y_train)
# 6. 최적 파라미터 및 성능 확인
print("Best Parameters:", grid_search.best_params_)
print("Best CV Score:", grid_search.best_score_)
# 7. 테스트 데이터 성능 확인
best_model = grid_search.best_estimator_
y_pred = best_model.predict(X_test)
test_acc = accuracy_score(y_test, y_pred)
print("Test Accuracy:", test_acc)
- 데이터 로드를 진행합니다.
- load_iris 함수를 사용해 붓꽃 데이터를 가져옵니다.
- 특징 행렬 X와 타깃 벡터 y를 분리합니다.
- 학습/테스트 데이터 분할을 합니다.
- train_test_split을 통해 데이터의 20%를 테스트 세트로 분할합니다.
- random_state=42로 설정하여 재현성이 있게 합니다.
- stratify=y를 통해 클래스 비율이 유지되도록 합니다.
- 하이퍼 파라미터 후보군을 정의합니다.
- n_estimators는 랜덤 포레스트가 생성할 결정 트리의 개수입니다.
- max_depth는 각 결정 트리의 최대 깊이로, None이면 제한 없이 분기할 수 있습니다.
- GridSearchCV를 생성합니다.
- param_grid에 정의한 파라미터 후보군을 넣어줍니다.
- cv=5로 5-Fold 교차검증을 수행합니다.
- scoring='accuracy'로 정확도를 기준으로 최적 파라미터를 찾습니다.
- n_jobs=-1로 가능한 모든 CPU 코어를 사용해 계산을 병렬화합니다.
- GridSearchCV로 학습합니다.
- fit(X_train, y_train)을 통해 주어진 파라미터 후보군 각각에 대해 모델을 학습하고 교차검증 성능을 비교합니다.
- 내부적으로 5-Fold 교차검증을 통해 파라미터 조합별 성능을 평가합니다.
- 최적 파라미터와 교차검증 성능을 확인합니다.
- best_params_는 최고 성능을 낸 파라미터 조합입니다.
- best_score_는 교차검증에서의 최고 정확도입니다.
- 테스트 세트로 최종 성능을 확인합니다.
- best_estimator_는 최적 파라미터로 학습된 모델을 의미합니다.
- 테스트 세트 예측 결과로 정확도를 측정하여 모델의 최종 성능을 평가합니다.
4. 머신러닝 추가 개념
- 최적화 (Oprimization)
- 하이퍼파라미터 튜닝 (GridSearchCV, RandomizedSearchCV 등)
- 피처 엔지니어링 (새로운 파생 변수 생성, 불필요한 변수 제거)
- 과적합 방지 (교차검증, 규제 적용, 드롭 아웃 등)
- 배포 (Deployment)
- 학습 완료 모델을 운영 환경에 배포
- API 서버 구축, 클라우드 (AWS, GCP) 또는 엣지 디바이스 (임베디드 환경)
- 지속적 모니터링으로 모델 성능이 저하될 경우 재학습 주기 설정
- MLOps(머신러닝 운영)
- Machine Learning + DevOps의 합성어
- 머신러닝 모델 개발부터 배포, 모니터링, 재학습, 롤백(Rollback) 등 전 과정을 자동화하고 효율적으로 운영하는 방법론
- 프로젝트 완성 → 실제 운영 단계에서 지속적인 모니터링과 데이터/모델 업데이트가 필요
5. 모델 해석 가능성 (Explainable AI, XAI)
- 머신러닝, 특히 딥러닝 모델은 블랙박스처럼 동작
- 의료/금융 등 규제 산업에서는 “왜 이런 결과가 나왔는지”에 대한 설명 요구
- 주요기법
- LIME(Local Interpretable Model-agnostic Explanations)
- SHAP(Shapley Additive Explanations)
- Feature Importance 시각화(트리 기반 모델)
- 코드예시
import matplotlib.pyplot as plt
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn.ensemble import RandomForestClassifier
# 1. 데이터 로드
iris = load_iris()
X = iris.data
y = iris.target
feature_names = iris.feature_names
# 2. 학습/테스트 데이터 분할
X_train, X_test, y_train, y_test = train_test_split(
X, y,
test_size=0.2,
random_state=42,
stratify=y
)
# 3. 랜덤 포레스트 모델 학습
rf = RandomForestClassifier(random_state=42)
rf.fit(X_train, y_train)
# 4. 피처 중요도 추출
importances = rf.feature_importances_
# 5. 시각화
plt.bar(range(len(importances)), importances)
plt.xticks(range(len(importances)), feature_names, rotation=45)
plt.xlabel("Feature")
plt.ylabel("Importance")
plt.title("Feature Importances in RandomForest")
plt.tight_layout()
plt.show()
# 가장 중요한 변수
most_important_idx = importances.argmax()
most_important_feature = feature_names[most_important_idx]
print("가장 중요한 변수:", most_important_feature)
- 데이터 로드를 진행합니다.
- load_iris 함수를 사용해 붓꽃 데이터를 불러옵니다.
- 붓꽃 데이터에는 총 4개의 특성이 있으며, 꽃받침과 꽃잎의 길이·너비 정보가 들어있습니다.
- 학습/테스트 데이터 분할을 합니다.
- train_test_split 함수를 이용하여 훈련용 데이터와 테스트용 데이터를 8:2 비율로 나누고, random_state=42로 설정하여 결과가 재현 가능하도록 합니다.
- 랜덤 포레스트(RandomForestClassifier) 모델을 생성하고 학습합니다.
- rf = RandomForestClassifier(random_state=42)로 모델을 생성합니다.
- fit(X_train, y_train)을 통해 훈련 세트로 모델을 학습합니다.
- **피처 중요도(feature_importances_)**를 추출합니다.
- 학습된 랜덤 포레스트 모델의 feature_importances_ 속성은 각 특성이 분류 작업에 얼마나 기여했는지 중요도를 나타냅니다.
- 시각화를 진행합니다.
- matplotlib의 bar 함수를 사용하여 중요도를 막대 그래프로 표시합니다.
- xticks에 feature_names를 연결해 어떤 막대가 어떤 특성에 해당하는지 확인할 수 있도록 합니다.
- 가장 중요한 변수를 찾습니다.
- importances.argmax()를 통해 가장 중요도가 높은 특성의 인덱스를 구하고, 해당 인덱스에 해당하는 특성 이름을 출력합니다
<총정리>
1. 앙상블 : 배깅과 부스팅을 통해 모델 조합으로 성능 향상
* 높은 예측 성능이 필요하고, 데이터 패턴이 복잡하거나 단일 모델로는 성능이 충분하지 않을 때 사용합니다. 예측 정확도를 최대화해야 하는 프로젝트(예: 금융 사기 탐지, 의료 진단)에서 자주 쓰이는 방법
- 배깅은 여러 모델을 ‘병렬적으로’ 독립 학습시킨 뒤 결과를 결합(평균/투표)하는 방식
- 부스팅은 모델을 ‘순차적으로’ 학습하면서 이전 모델의 오류를 단계적으로 보정해 나가는 방식
2. 과적합 vs 과소적합
- 과적합 : 복잡도 높음 → 정규화, 드롭아웃, 조기 종료, 데이터 증강 등으로 대응
- 과소적합 : 복잡도 낮음 → 모델 파라미터 수 늘리기, 학습 기간(에폭) 늘리기
3. 하이퍼파라미터 튜닝 : GridSearchCV, RandomizedSearchCV 등을 통해 모델 성능 최적화
- 탐색해야 할 파라미터 범위가 좁고, 후보가 적을 때는 GridSearchCV가 효율적
- 후보 범위가 매우 넓고 많은 경우에는 RandomizedSearchCV로 무작위 샘플링을 활용하여 탐색 시간을 줄이는 것이 좋음
4. Feature Importance 시각화를 통해 트리기반의 모델의 경우 어느정도 해석도 가능
[실습 문제]
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score, confusion_matrix, classification_report
# 1. 유방암 데이터셋 로드
data = load_breast_cancer()
X = data.data
y = data.target
feature_names = data.feature_names
- 유방암 데이터셋을 사이킷런에서 불러오세요.
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score, confusion_matrix, classification_report
# 1. 유방암 데이터셋 로드
data = load_breast_cancer()
X = data.data
y = data.target
feature_names = data.feature_names
- 학습용/테스트용 데이터로 분할하세요.
# 2. 학습용 / 테스트용 분할
X_train, X_test, y_train, y_test = train_test_split(X, y,
test_size = 0.2,
random_state = 42,
stratify = y)
print(X_train.shape) # (455, 30)
print(y_train.shape) # (455,)
print(X_test.shape) # (114, 30)
print(y_test.shape) # (114,)
- XGBoost 분류 모델을 생성하여 학습하세요.
- random_state 를 지정해야 하는 경우 42로 지정하세요.
# 3. XGBoost 분류 모델 생성
from xgboost import XGBClassifier
from sklearn.metrics import accuracy_score, confusion_matrix, classification_report
xgb_model = XGBClassifier(random_state = 42)
xgb_model.fit(X_train, y_train)
y_pred_xgb = xgb_model.predict(X_test)
- 테스트 세트로 예측을 수행하고, 정확도(Accuracy), 혼동 행렬(Confusion Matrix), 분류 보고서(Classification Report) 등을 구해 성능 평가를 하세요.
# 4. 정확도, 혼동 행렬, 분류 보고서
print("===XGBoost===")
print("Accuracy:", accuracy_score(y_test, y_pred_xgb))
print("Confusion Matrix:\n", confusion_matrix(y_test, y_pred_xgb))
print("Classification Report:\n", classification_report(y_test, y_pred_xgb))
- 피처 중요도를 시각화한 뒤, 그중 가장 중요한 변수가 무엇인지 찾아 출력하십시오.
# 5. 피처 중요도 시각화 & 가장 중요한 변수 찾기
importance = xgb_model.feature_importances_
plt.bar(range(len(importance)), importance)
plt.xticks(range(len(importance)),feature_names, rotation=90)
plt.xlabel('Feature')
plt.ylabel('Importance')
plt.title("Feature Importance in XGBoost")
plt.show()