返回首页>>

基于随机森林的收入预测案例





案例整理者:
李山雨
最后修订:
2021-3-3

任务类型:预测问题

算法:随机森林算法

数据源:income_evaluation(3.7M)

下载地址1(income_evaluation.csv ) 下载地址2(北大网盘


      简介:本文拟对kaggle上的一篇关于随机森林模型的文章进行案例调研,对文章的各个环节,包括案例描述、数据检视、特征工程、模型训练和模型优化使用的方法进行梳理并对文章的亮点与不足做出相应的反思。该文章的任务是基于随机森林模型,结合人的15项特征来预测该人年收入是否超过5万美元。数据源是美国人口普查局1994年的普查数据,调查对象数量为32561。这篇文章的亮点有:1.特征工程使用了简单而且容易理解的方法;2.进行充分的数据检视,对后续数据处理有了启示性作用;3、通过计算特征重要度过滤不重要的特征,展现了模型可以不断优化的可能性。主要不足有:1、没有解决高基数问题;2.并没有优化模型参数;3.没有处理离群值。

案例分析文档:基于随机森林的收入预测.docx(2.0M)

案例分析展示:基于随机森林的收入预测.pptx(1.8M)

案例分析源代码:income_predict.py(4K)