决策树案例

第1关:基于决策树模型的应用案例

任务描述
本关任务:使用决策树算法完成成人收入预测。

相关知识
为了完成本关任务,你需要掌握:1.数据特征处理,2.使用决策树算法完成成人收入预测。

数据处理及特征工程
本次任务我们将会使用成人数据集(来源于UCI数据集:http://archive.ics.uci.edu/ml/datasets/Adult ),该数据集从美国1994年人口普查数据库中抽取而来,共48842条数据,涉及到收入统计,因此也称作“人口普查收入”数据集。该数据集可以通过对属性变量包括年龄、工种、学历、职业等14种信息来判断年收入是否超过50k。
首先,我们来了解一下我们的数据,从而进一步明确任务,并作出针对性的分析。人口普查收入数据集是一个分类问题,目的是用来预测年收入是否超过50k$,对应我们所需要的标签,这在数据集中以“<=50K”和“>50K”来进行表示;除此之外,数据包含了用于判断该决定的相关信息,对应我们所需要的特征,这些特征一共有14个。

编程要求
根据提示,在右侧编辑器补充代码,实现使用决策树进行成人收入预测,包括:

数据处理
模型训练
模型预测
测试指标计算,并打印aou的值,函数:
print(“auc的值:{}”.format(auc))
测试说明
平台会对你编写的代码进行测试:

预期输出:

提示:
参照示例完成任务

开始你的任务吧,祝你成功࿰

你可能感兴趣的:(Educoder实训,决策树,机器学习,python)