python调包侠_sklearn调包侠之决策树算法

本文介绍了决策树算法的基本原理和改进算法,如C4.5和CART。通过泰坦尼克号生还数据集进行实战,展示了数据预处理、模型训练与评估。使用sklearn的DecisionTreeClassifier,通过调整参数解决过拟合问题,如设置max_depth、min_samples_split等。最后,通过网格搜索GridSearchCV进行模型调优,找到最佳参数组合。
摘要由CSDN通过智能技术生成

8aee00ab196d

决策树原理

改进算法

但使用信息增益作为特征选择指标(ID3算法)容易造成过拟合。举一个简单例子,每个类别如果都有一个唯一ID,通过ID这个特征就可以简单分类,但这并不是有效的。为了解决这个问题,有了C4.5和CART算法,其区别如下所示:

ID3 是信息增益划分

C4.5 是信息增益率划分

CART 做分类工作时,采用 GINI 值作为节点分裂的依据

实战——泰坦尼克号生还预测

数据导入与预处理

该数据可在kaggle网站下载,这里我们先通过pandas读入数据。

import numpy as np

import pandas as pd

df = pd.read_csv('data/titanic/train.csv',index_col=0)

df.head()

8aee00ab196d

首先,对于一些不重要的信息进行删除(例如Name);我们都知道,机器学习是没法对字符串进行计算的,这里需要把Sex、Embarked转换为整数类型。

# 删除列

df.drop(['Name', 'Ticket',

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值