728x90
XGBClassifier
- XGBClassifier에 대한 GPT의 답변
- params 대입 방법
XGBClassifier에 대한 GPT의 답변
params 대입 방법
# 라이브러리 불러오기
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split
# 데이터 불러오기
data = pd.read_csv(path)
# target 확인
target = 'ADMIT'
# 데이터 분리
x = data.drop(target, axis=1)
y = data[target]
# 7:3으로 분리
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.3, random_state=1)
# 선언하기
model = XGBClassifier(max_depth=5, random_state=1)
# 학습하기
model.fit(x_train,y_train)
# 예측하기
y_pred = model.predict(x_test)
# 평가하기
print(confusion_matrix(y_test, y_pred))
print(classification_report(y_test,y_pred))
728x90
'데이터 - 머신러닝 지도 학습' 카테고리의 다른 글
지도 학습] 회귀 KNeighborsRegression (0) | 2023.09.23 |
---|---|
지도 학습] 분류 - LogisticRegression (0) | 2023.09.21 |
지도 학습] 모델 저장 (0) | 2023.09.21 |
지도 학습] 분류 - RandomForestClassifier 랜덤 포레스트 (0) | 2023.09.20 |
지도 학습] 분류 - 결정트리 DecisionTreeClassifier (0) | 2023.09.20 |