scikit-learnの決定木

scikit-learnの決定木を試してみました。
データは先日と同様のWikipediaのものを使用です。
データの中身は数値にしないといけないみたいなので、適当に変換してみました。

 # coding: utf8
from sklearn import tree

data = [ [0,0,0],
    [0,1,0],
    [1,1,0],
    [0,0,1],
    [1,0,0] ]

target = ['鳥類', '哺乳類' , '哺乳類' , '爬虫類' , '鳥類']

clf = tree.DecisionTreeClassifier()
clf = clf.fit(data, target)

tree.export_graphviz(clf, out_file='test.dot',
    feature_names=range(3), class_names=sorted(list(set(target))))