首页 > 其他 > 详细

决策树案例(红酒)

时间:2021-07-07 12:39:27      阅读:27      评论:0      收藏:0      [点我收藏+]
# 决策树
from sklearn import tree
from sklearn.datasets import load_wine
from sklearn.model_selection import train_test_split
import pandas as pd
wine = load_wine()
X=wine.data
Y=wine.target
df = pd.DataFrame(X)
df["class"] = Y

clf = tree.DecisionTreeClassifier(criterion="entropy")

x_train,x_test,y_train,y_test = train_test_split(X,Y,test_size=0.3)
clf.fit(x_train,y_train)
score = clf.score(x_test,y_test)
print(score)
import graphviz
feature_name = wine["feature_names"]
dot_data = tree.export_graphviz(clf,
    feature_names = feature_name,
    filled=True,
    rounded=True
)
graph=graphviz.Source(dot_data)
print(graph)

 

决策树案例(红酒)

原文:https://www.cnblogs.com/djwww/p/14980432.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!