说明:这是一个机器学习实战项目(附带数据+代码+文档+视频讲解),如需数据+代码+文档+视频讲解可以直接到文章最后获取。

1.项目背景

 CatBoost是一种基于对称决策树(oblivious trees)为基学习器实现的参数较少、支持类别型变量和高准确性的GBDT框架,主要解决的痛点是高效合理地处理类别型特征,这一点从它的名字中可以看出来,CatBoost是由Categorical和Boosting组成。

Optuna是一个开源的超参数优化(HPO)框架,用于自动执行超参数的搜索空间。 为了找到最佳的超参数集,Optuna使用贝叶斯方法。

本项目使用基于Optuna超参数自动优化的CatBoostClassifier算法来解决分类问题。

2.数据获取

本次建模数据来源于网络(本项目撰写人整理而成),数据项统计如下:

数据详情如下(部分展示):

3.数据预处理

3.1 用Pandas工具查看数据

使用Pandas工具的head()方法查看前五行数据:

从上图可以看到,总共有8个字段。

关键代码:

3.2 缺失值统计

使用Pandas工具的info()方法统计每个特征缺失情况:

从上图可以看到,数据不存在缺失值,总数据量为1000条。

关键代码:

3.3 变量描述性统计分析

通过Pandas工具的describe()方法来来统计变量的平均值、标准差、最大值、最小值、分位数等信息:

 关键代码如下:

4.探索性数据分析

4.1 y变量分类柱状图

用Pandas工具的value_counts().plot()方法进行统计绘图,图形化展示如下:

从上面两个图中可以看到,分类为0和1的样本,数量基本一致。

4.2 y变量类型为0 x1变量分布直方图

通过Matpltlib工具的hist()方法绘制直方图:

从上图可以看出,x1主要集中在-1到2之间。

4.3 相关性分析

通过Pandas工具的corr()方法和seaborn工具的heatmap()方法绘制相关性热力图:

从图中可以看到,正数为正相关,负数为负相关,绝对值越大相关性越强。

5.特征工程

5.1 建立特征数据和标签数据

y为标签数据,除 y之外的为特征数据。关键代码如下:

 

5.2 数据集拆分

数据集集拆分,分为训练集和测试集,80%训练集和20%测试集。关键代码如下:

6.构建Optuna超参数自动化的CatBoost分类模型

主要使用基于Optuna超参数自动化调优的CatBoostClassifier算法,用于目标分类。

6.1 Optuna超参数自动化调优框架介绍

Optuna是一个开源的超参数优化(HPO)框架,用于自动执行超参数的搜索空间。 为了找到最佳的超参数集,Optuna使用贝叶斯方法。 它支持下面列出的各种类型的采样器:

  1. GridSampler (使用网格搜索)
  2. RandomSampler (使用随机采样)
  3. TPESampler (使用树结构的Parzen估计器算法)
  4. CmaEsSampler (使用CMA-ES算法)

一个极简的 Optuna 的优化程序中只有三个最核心的概念,目标函数(objective),单次试验(trial),和研究(study):

  1. objective 负责定义待优化函数并指定参/超参数数范围
  2. trial 对应着 objective 的单次执行
  3. study 则负责管理优化,决定优化的方式,总试验的次数、试验结果的记录等功能。

6.2 构建调优模型

 关键代码如下:

6.3 最优参数展示

最优参数结果展示:

关键代码如下:

7.模型评估

7.1 评估指标及结果

评估指标主要包括准确率、查准率、召回率、F1分值等等。

从上表可以看出,基于Optuna超参数自动调优的CatBoost分类模型效果良好。

关键代码如下:

7.2 分类报告

基于Optuna超参数自动调优的CatBoost分类模型分类报告:

从上图可以看到,分类类型为0的F1分值为0.92;分类类型为1的F1分值为0.93;整个模型的准确率为0.93。

8.结论与展望

综上所述,本项目采用了基于Optuna超参数自动调优的CatBoost分类模型,最终证明了我们提出的模型效果良好。

# 本次机器学习项目实战所需的资料,项目资源如下:
 
# 项目说明:

# 链接:https://pan.baidu.com/s/1xVg3Dm0XEf3dBfhrw-CdrQ 
# 提取码:uctg

# 用Pandas工具查看数据
print(data.head())

# 数据缺失值统计
print('****************************************')
print(data.info())

print('****************************************')
print(data.describe().round(4))  # 保留4位小数点

# y变量分类柱状图
plt.rcParams['font.sans-serif'] = ['SimHei']  # 指定默认字体
plt.rcParams['axes.unicode_minus'] = False  # 解决保存图像是负号'-'显示为方块的问题
# kind='bar' 绘制柱状图
data['y'].value_counts().plot(kind='bar')
plt.xlabel("y变量类型")
plt.ylabel("数量")
Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐