python决策树 sklearn鸢尾花数据集分类

def decision_iris():
    """用决策树进行鸢尾花分类"""
    #导入数据
    iris = load_iris()
    #划分数据
    x_train,x_test,y_train,y_test = train_test_split(iris.data,iris.target,random_state=6)
    #决策树预估器
    estimator = DecisionTreeClassifier(criterion= "entropy")
    estimator.fit(x_train,y_train)
    #模型评估
    # 1.直接比对真实目标值与预估目标值
    y_predict = estimator.predict(x_test)
    print("预估值：\n", y_predict)
    print(y_predict == y_test)  # 真实目标值与预估值比对
    # 2.计算准确率
    score = estimator.score(x_test, y_test)
    print("准确率：\n", score)
    #决策树可视化
    export_graphviz(estimator,out_file="iris_tree.dot",feature_names= iris.feature_names)