import pydotplus
from sklearn.tree import DecisionTreeClassifier
from sklearn import datasets
from IPython.display import display, Image
from sklearn import tree
//使用官方数据集iris
iris = datasets.load_iris()
features = iris.data
target = iris.target
//构建分类器
decision_tree = DecisionTreeClassifier(random_state=0)
//训练模型
model = decision_tree.fit(features, target)
//可视化
dot_data = tree.export_graphviz(decision_tree,
out_file=None,
feature_names=iris.feature_names,
class_names=iris.target_names)
graph = pydotplus.graph_from_dot_data(dot_data)
graph.write_png('iris.png')
机器学习练习——使用决策树实现iris分类器
最新推荐文章于 2024-04-16 03:16:26 发布