본문 바로가기
머신러닝/6. 의사결정나무

(p3)제04강_01 파이썬으로 해 보는 의사결정나무

by 통컨이 2020. 11. 10.
728x90

import numpy as np

from sklearn import datasets # iris 데이터를 불러오기

from sklearn import tree # 의사결정나무 모듈

from sklearn.model_selection import train_test_split # 훈련데이터와 테스트 데이터 분류할 때 필요

from sklearn.preprocessing import StandardScaler # 표준화할 때 필요

iris = datasets.load_iris() # iris 데이터 로딩

X = iris.data

y = iris.target

model_tree = tree.DecisionTreeClassifier(criterion='entropy', max_depth=3, random_state=0)

model_tree.fit(X, y)

model_tree.score(X, y)