확보한 데이터 중에서 모델 학습에 사용하지 않고 빼둔 데이터를 가지고 모델을 테스트한다
from sklearn.datasets import load_iris # sklearn에서 제공하는 iris 데이터
import pandas as pd
iris = load_iris()
평면에 그리려고 데이터 2개만 사용
from sklearn.model_selection import train_test_split
features = iris.data[:, 2:]
labels = iris.target
# test_size=0.2 -> train data 80%, test data 20%
X_train, X_test , y_train, y_test = train_test_split(features, labels, test_size=0.2, random_state=13)
X_train.shape, X_test.shape
iris.target_names
iris 데이터로 생각하면 'setosa', 'versicolor', 'virginica'가 각 각 50개씩 있는 150개 데이터라 생각할 때 train 데이터와 test 데이터에 각각 'setosa', 'versicolor', 'virginica' 비율이 어떻게 되는지 확인 해야한다.
import numpy as np
np.unique(y_test, return_counts=True)
9, 8, 13개로 테스트 데이터가 구성되었다.
테스트 데이터가 각 각 10개 10개 10개 라면 좋을 텐데 9, 8, 13개로 구성되어있다.
이것이 문제인가 생각했을 땐 no! 엔지니어 판단에 달려있다.
But! 일반적으로는 구성 비율을 같게한다. (stratify 사용)
X_train, X_test , y_train, y_test = train_test_split(features, labels,
test_size=0.2,
stratify=labels,
random_state=13)
X_train.shape, X_test.shape
np.unique(y_test, return_counts=True)
train 데이터 대상으로 DecisionTree 만들기
from sklearn.tree import DecisionTreeClassifier
iris_tree = DecisionTreeClassifier(max_depth=2, random_state=13)
iris_tree.fit(X_train, y_train)
import matplotlib.pyplot as plt
from sklearn.tree import plot_tree
# iris_tree -> 학습된 모델
# 학습된 모델의 구조 구현
plot_tree(iris_tree);
from sklearn.metrics import accuracy_score
# iris_tree -> 학습이 완료된 상태
# iris.data[:, 2:] 데이터를 줄테니 맞춰봐(predict) -> 지금은 정답을 주지 않음
y_pred_tr = iris_tree.predict(iris.data[:, 2:])
# 원래의 참값과, 학습된 결과의 예측값의 정확도
accuracy_score(iris.target, y_pred_tr)
train 데이터에 대한 결정 경계 확인
from mlxtend.plotting import plot_decision_regions
plt.figure(figsize=(14, 8))
plot_decision_regions(X=X_train, y=y_train, clf=iris_tree, legend=2)
plt.show()
iris는 원래 %가 높게 나옴
지금은 max_depth를 한정하는 것이 좋구나, 데이터를 나누는 것이 좋구나 를 과적합을 막기 위해 생각하는 훈련
내가 가진 train 데이터에서 95%가 나왔으니 test 데이터(일반적인)에서 확인하는 작업 필요
y_pred_test = iris_tree.predict(X_test)
accuracy_score(y_test, y_pred_test)
test 데이터 accuracy가 train 데이터 accuracy보다 높아지는 경우는 거의 없다. (iris 데이터라서 test 데이터 accuracy가 높게 나왔다.)
<생각하는 방법>
- 'train accuracy에서 95%가 나왔는데 내가 가진 test accuracy에서 비슷한(근처) %가 나왔으니 일단 과적합은 안일어났구나' 라고 생각
- 아래 그래프에서 한 개가 틀렸다는 중요하지 않다!!
plt.figure(figsize=(14, 8))
plot_decision_regions(X=X_test, y=y_test, clf=iris_tree, legend=2)
plt.show()
scatter_highlight_kwargs = {'s':150, 'label':'Test data', 'alpha':0.9} # 강조하는 데이터 -> 사이즈, 라벨, 투명도
scatter_kwargs = {'s':120, 'edgecolor':None, 'alpha':0.9} # 일반적 120개 데이터 -> 크기를 작게하고 테두리 선 표시 안함
plt.figure(figsize=(12,8))
plot_decision_regions(X=features, y=labels,
X_highlight=X_test, clf=iris_tree, legend=2,
scatter_highlight_kwargs=scatter_highlight_kwargs,
scatter_kwargs=scatter_kwargs,
contourf_kwargs={'alpha':0.2});
위에서 2개로 해봤으니 4개로 구해보자
features = iris.data
labels = iris.target
X_train, X_test , y_train, y_test = train_test_split(features, labels,
test_size=0.2,
stratify=labels, # labels을 기준으로 구성비율을 맞춘다
random_state=13)
iris_tree = DecisionTreeClassifier(max_depth=2, random_state=13)
iris_tree.fit(X_train, y_train)
plot_tree(iris_tree);
y_pred_tr = iris_tree.predict(iris.data)
# 원래의 참값과, 학습된 결과의 예측값의 정확도
accuracy_score(iris.target, y_pred_tr)
y_pred_test = iris_tree.predict(X_test)
accuracy_score(y_test, y_pred_test)
test_data = [[4.3, 2. , 1.2, 1.0 ]]
iris_tree.predict_proba(test_data)
test_data = [[4.3, 2. , 1.2, 1.0 ]]
iris.target_names[iris_tree.predict(test_data)]
즉 test_data가 versicolor일 확률이 97%이다.
iris_tree.feature_importances_
iris_tree = DecisionTreeClassifier(max_depth=5, random_state=13)
iris_tree.fit(X_train, y_train)
iris_tree.feature_importances_
iris_tree = DecisionTreeClassifier(max_depth=2, random_state=13)
iris_tree.fit(X_train, y_train)
# 중요도
dict(zip(iris.feature_names, iris_tree.feature_importances_))
리스트 > 튜플
li1 = ['a', 'b', 'c']
li2 = [1, 2, 3]
pairs = [pair for pair in zip(li1, li2)]
pairs
dict(pairs)
dict(zip(li1, li2))
위의 작업을 푸는 작업
x, y = zip(*pairs)
print(x)
print(y)
print(list(x))
print(list(y))
"이 글은 제로베이스 데이터 취업 스쿨 강의를 듣고 작성한 내용으로 제로베이스 데이터 취업 스쿨 강의 자료 일부를 발췌한 내용이 포함되어 있습니다."