gini函数和entropy 函数 数据分类效果对比

gini函数和entropy 函数 数据分类效果对比
from sklearn.datasets import load_iris
from sklearn import tree
import os
import pydot
print(os.getcwd())
#clf = tree.DecisionTreeClassifier(criterion = “entropy”) #entropy
clf = tree.DecisionTreeClassifier(criterion = “gini”) #gini
iris = load_iris()
clf = clf.fit(iris.data, iris.target)
tree.export_graphviz(clf, out_file=‘0_DT_Tree/tree1.dot’)
(graph,) = pydot.graph_from_dot_file(‘0_DT_Tree/tree1.dot’)
graph.write_png(‘0_DT_Tree/tree1.png’)