如何找到BlackBox模型的特征重要性?

新知榜官方账号

2023-08-21 16:28:43

数据科学家Rahul Agarwal介绍特征选择方法——排列重要性

数据科学是对算法进行的研究。每天我们都要处理很多的算法,因此需要列出一些最常见和最常用的算法,这些算法将会在新的DS算法系列中使用。

当我们创建了很多特征的时候,我们常常需要想尽方法减少特征的数量。上次我写了一篇名为“每个数据科学家都应该知道的5种特征选择算法”的文章,其中谈到了使用相关性或基于树类结构的方法,并在特征选择的过程中添加了一些结构。最近我接触了另一种新颖的特征选择方法,叫做排列重要性。这篇文章将介绍排列重要性是如何起作用的以及如何使用ELI5对其进行编码。

什么是排列重要性

简单地说,如果我们从数据集中删除一个特定的特征,我们可以根据我们的评估指标(F1、精度AUC等)的变化来给这个特征赋予重要性。要做到这一点可就太简单了,我们从数据集中删除了一个特征,然后训练分类器,接着查看评估指标是如何变化的。我们要对所有的特征都这样操作。所以我们至少要拟合n次模型,其中n是模型中的特征的数量。这意味着非常多的计算和工作量。我们是否能在这样的条件下做到更好呢?

为了计算排列的重要性,我们对单个特征的值进行改组、排列,并使用所得的数据集进行预测。然后将这些预测用于计算我们的评估指标。直观地讲,如果我们的准确性或任何评估指标都没有受到影响,那我们就可以说这个特征并不重要。但如果我们的准确性受到了影响,那我们也得承认该特征确实很重要。

如何使用排列重要性

我们使用更新的数据集对特征进行排序和预测。直观地讲,如果我们的准确性或任何评估指标都没有受到影响,那我们就可以说这个特征并不重要。但如果我们的准确性受到了影响,那我们也得承认该特征确实很重要。

我们可以将排列重要性应用于sklearn模型。首先,我们要开始训练模型。然后使用eli5sklearn模块中的PermutationImportance函数。我们还可以使用eli5来计算非机器学习模型的特征重要性。在这里我们训练了一个LightGBM模型。我们需要为评分函数创建一个包装器,以计算我们的评估指标。现在我们可以使用eli5.permutation_importance中的get_score_importances函数以获得最终的特征函数。

特征工程和特征选择是任何机器学习通道的关键部分。我们追求模型的准确性,如果不反复查看这些模型,就不可能获得良好的准确性。在这篇文章中,我试图将排列重要性解释为一种特征选择方法。它可以帮助我们找到任何BlackBox模型的特性重要性,这与我之前关于特征选择的文章中的技术表达不同。

本页网址:https://www.xinzhibang.net/article_detail-10155.html

寻求报道,请 点击这里 微信扫码咨询

关键词

特征选择 排列重要性 ELI5

分享至微信: 微信扫码阅读

相关工具

相关文章