当前位置 博文首页 > Weiyaner的博客:机器学习项目实战:基于随机森林进行心脏病分类
??本项目是Kaggle上面的一个经典竞赛题,心脏病分类问题,题目链接在这里. 主要基于随机森林的bagging集成学习框架,通过13个生理特征数据,实现对心脏病分类的预测。
??由于自己想要在这个项目更多的学习到模型解释方面的内容,所以对于模型精度没有过多的在意和调参。模型解释主要用了eli5,shap和部分依赖图。
??下面是完整的代码和运行结果。在python3.7环境下可以运行。