XGBOOST 초간단 회귀분석 사용법¶
- 초간단 최소한의 훈련 방법
In [22]:
import xgboost
import pandas as pd
import seaborn as sns
데이터로 변환¶
In [12]:
raw = pd.read_csv("data.csv")
raw.head()
Out[12]:
독립, 종속변수 나눔¶
In [14]:
X = raw[['KM','AGE']]
y = raw[['PRICE']]
모델 선언¶
In [58]:
model = xgboost.XGBRegressor(learning_rate=0.1,
max_depth=5,
n_estimators=100)
모델 훈련¶
In [18]:
model.fit(X,y)
Out[18]:
훈련결과 확인¶
In [57]:
sns.scatterplot(raw['PRICE'], model.predict(X))
Out[57]: