더북(TheBook)

앞에서 스케일링한 데이터셋에 결정 트리를 적용하면 다음과 같이 구현할 수 있습니다.

실습 | 결정 트리 분류기로 학습하기

# 결정 트리 분류기를 불러옵니다.
from sklearn.tree import DecisionTreeClassifier ----- ➊

# 학습 환경을 설정합니다.
classifier = DecisionTreeClassifier() ----- ➋
classifier.fit(X_train, y_train) ----- ➌
  
# 테스트셋에 적용합니다. 
y_pred = classifier.predict(X_test) ----- ➍

# 계층별 교차 검증 환경을 설정합니다. 
skf = StratifiedKFold(n_splits=10, shuffle=True)

# 교차 검증을 통해 정확도를 계산합니다. 
accuracies = cross_val_score(estimator=classifier, X=X_train, y=y_train, cv=skf)

# 정확도와 표준편차를 출력합니다. 
print("Accuracy: {:.2f} %".format(accuracies.mean()*100))
print("Standard Deviation: {:.2f} %".format(accuracies.std()*100))
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.