机器学习-决策树实现可视化

机器学习- 决策树实现可视化

1. 要安装库sklearn，pandas，graphviz在cmd中安装
   pip install sklearn
   pip install pandas
   pip install graphviz

2. 下载graphviz，装环境变量
   下载网址：在下载的时候有点慢https://graphviz.gitlab.io/_pages/Download/Download_windows.html
   安装环境变量：
   在我的环境变量里添加你下载包的位置
   
   然后在系统环境变量里添加
   
   

3. 简单代码块

import pandas as pd
from sklearn.tree import DecisionTreeClassifier,export_graphviz
from sklearn.metrics import classification_report
import graphviz

data = pd.read_csv("titanic_data.csv") #将csv导入

data.drop("PassengerId",axis = 1,inplace = True)#删除pessengerId，inplace=true表示在data中直接改

data.loc[data['Sex'] == 'male','Sex'] = 1 #将数值1来代替male，用0来代替female
data.loc[data['Sex'] =&