ML(0813_day8) - 결정 트리

결정 트리¶

In [3]:

import numpy as np
import matplotlib.pyplot as plt

from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier

In [4]:

iris = load_iris()

X = iris.data[:, 2:] # petal length, petal width
y = iris.target

In [8]:

tree_clf = DecisionTreeClassifier(max_depth=2, random_state=42)
tree_clf.fit(X, y)

Out[8]:

DecisionTreeClassifier(max_depth=2, random_state=42)

In [9]:

from sklearn.tree import export_graphviz

In [13]:

export_graphviz(tree_clf, 
                out_file="iris_tree.dot", 
                class_names=iris.target_names, 
                feature_names=iris.feature_names[2:], 
                rounded=True, 
                filled=True)

In [14]:

import graphviz

with open("iris_tree.dot") as f:
    dot_graph = f.read()
graphviz.Source(dot_graph)

Out[14]:

In [ ]: