贪心科技机器学习训练营（十）

先把来源写上

来源：贪心学院，https://www.zhihu.com/people/tan-xin-xue-yuan/activities

使用决策树预测糖尿病 分类任务
数据源: https://www.kaggle.com/uciml/pima-indians-diabetes-database#diabetes.csv

注意: 需要使用如下命令安装额外两个包用于画图

conda install python-graphviz

conda install pydotplus

  

 

  
1
  
2
  
3
 

前面的不说了

顺便画下相关图

import seaborn as sns
from matplotlib import  pyplot as plt
plt.figure(figsize=(12,10))  # on this line I just set the size of figure to 12 by 10.
p=sns.heatmap(pima.corr(), annot=True,cmap ='RdYlGn')  # seaborn has very simple solution for heatmap

  

 

  
1
  
2
  
3
  
4
 

文章来源: maoli.blog.csdn.net，作者：刘润森！，版权归原作者所有，如需转载，请联系作者。

原文链接：maoli.blog.csdn.net/article/details/96470280