from sklearn.tree import DecisionTreeClassifier
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn import metrics
import pandas as pd
## print(load_iris())
X=load_iris().data
a=pd.DataFrame(X,columns=['sepal length','sepal width','petal length','petal length'])
## print(a)
y=load_iris().target
## print(X)
## print(y)
X_train,X_test,y_train,y_test=train_test_split(X,y,test_size=0.2,random_state=1)
D_tree=DecisionTreeClassifier()
D_tree.fit(X_train,y_train)
y_predict=D_tree.predict(X_test)
print(y_test)
print(y_predict)
print('accuracy:',metrics.accuracy_score(y_test,y_predict))
print('recall:',metrics.recall_score(y_test,y_predict,average='micro'))
print('precision:',metrics.precision_score(y_test,y_predict,average='micro'))
[0 1 1 0 2 1 2 0 0 2 1 0 2 1 1 0 1 1 0 0 1 1 1 0 2 1 0 0 1 2]
[0 1 1 0 2 1 2 0 0 2 1 0 2 1 1 0 1 1 0 0 1 1 2 0 2 1 0 0 1 2]
accuracy: 0.966666666667
recall: 0.966666666667
precision: 0.966666666667