咪咪色图
色中色电影 R话语fastshap
发布日期:2024-09-08 16:29    点击次数:118
 

色中色电影 R话语fastshap

图片

图片

图片

图片

图片

图片

图片

图片

SHAP(SHapley Additive exPlanations)是一种模子无关的讲明标准,因此它适用于任何模子。前边依然先容过屡次这个SHAP了,在R话语中杀青SHAP也长短常肤浅的。

R话语SHAP模子讲明R话语shapviz杀青SHAP可视化R话语SHAP模子讲明之kernelshap

fastshap基于Rcpp和foreach,是以它的速率绝顶快(比之前先容的几种齐快),何况fastshap既不错杀青局部讲明,又能杀青全局讲明,还不错无缝对接shapviz的可视化。

色情网站装配
# Install the latest stable version from CRAN:install.packages("fastshap")# Install the latest development version from GitHub:if (!requireNamespace("remotes")) {  install.packages("remotes")}remotes::install_github("bgreenwell/fastshap")
准备数据和R包

咱们使用经过缺失值插补的泰坦尼克号数据集t1,该数据有1309行,6列,其中survived是效果变量,二分类,1代表死字,2代表存活。

library(fastshap)t1 <- titanic_mice[[1L]]dim(t1)## [1] 1309    6str(t1)## 'data.frame':    1309 obs. of  6 variables:##  $ survived: Factor w/ 2 levels "no","yes": 2 2 1 1 1 2 2 1 2 1 ...##  $ pclass  : int  1 1 1 1 1 1 1 1 1 1 ...##  $ age     : num  29 0.92 2 30 25 48 63 39 53 71 ...##  $ sex     : Factor w/ 2 levels "female","male": 1 2 1 2 1 2 1 2 1 2 ...##  $ sibsp   : int  0 1 1 1 1 0 1 0 2 0 ...##  $ parch   : int  0 2 2 2 2 0 0 0 0 0 ...

把pclass变为有规则的因子型:

t1$pclass <- as.ordered(t1$pclass)  # makes more sense as an ordered factor

本次使用ranger建立当场丛林模子,对于其他模子,fastshap齐是营救的。

library(ranger)set.seed(2053)  # for reproducibility(rfo <- ranger(survived ~ ., data = t1, probability = TRUE))## Ranger result## ## Call:##  ranger(survived ~ ., data = t1, probability = TRUE) ## ## Type:                             Probability estimation ## Number of trees:                  500 ## Sample size:                      1309 ## Number of independent variables:  5 ## Mtry:                             2 ## Target node size:                 10 ## Variable importance mode:         none ## Splitrule:                        gini ## OOB prediction error (Brier s.):  0.1337358
局部讲明

为了证实若何最肤浅地使用Shapley值来量化特征孝敬,咱们需要一个新的不雅测值来预见。底下咱们将为新建一个不雅测值,给这个不雅测取个名字叫jack.dawson:

jack.dawson <- data.frame(  #survived = 0L,  # in case you haven't seen the movie  pclass = 3L,     # third-class passenger  age = 20.0,      # twenty years old  sex = factor("male", levels = c("female", "male")),  # male  sibsp = 0L,      # no siblings/spouses aboard  parch = 0L       # no parents/children aboard)

使用fastshap进行模子讲明和DALEX同样,领先亦然需要建立一个讲明器,用来领路模子和数据的各式信息。

在DALEX中,预见函数默许会从模子对象中索求(默许使用predict()),然则在fastshap中需要咱们自界说预见函数,用于告成复返数值(转头任务)概况类别概率(分类任务)。

# 自界说预见函数pfun <- function(object, newdata) {  # prediction wrapper  unname(predict(object, data = newdata)$predictions[, "yes"])}# 策画Jack's的生涯概率(jack.prob <- pfun(rfo, newdata = jack.dawson))## [1] 0.1314723

要是告成使用以下代码得到的并不是类别概率:

predict(rfo, jack.dawson)## Ranger prediction## ## Type:                             Probability estimation ## Sample size:                      1 ## Number of independent variables:  5

用这个函数不错得到统统t1不雅测的平均生涯概率:

# 策画统统东说念主的平均生涯概率(baseline <- mean(pfun(rfo, newdata = t1)))  ## [1] 0.3821045
# Difference between Jack and average(difference <- jack.prob - baseline)## [1] -0.2506322

不错发现Jack的生涯可能性低于平均值。底下咱们就使用SHAP来讲明为什么会这么。

底下就不错建立讲明器了,和DALEX同样,亦然使用explain()函数,亦然需要提供只含预见变量的数据框:

X <- subset(t1, select = -survived)  # 只含预见变量set.seed(2113)# 建立讲明器(ex.jack <- explain(rfo, X = X, pred_wrapper = pfun, newdata = jack.dawson))##      pclass          age sex      sibsp parch## [1,]      0 -0.005012306   0 0.02174902     0## attr(,"baseline")## [1] 0## attr(,"class")## [1] "explain" "matrix"  "array"

fastshap包使用高效版块的蒙特卡洛(Monte-Carlo,MC)算法。因此,为了清爽性和准确性,应屡次策画特征孝敬,并将效果取平均值。为此,只需将nsim参数缔造为一个相比大的值即可(默许是1)。底下咱们策画Jack的1000个基于Shapley的特征孝敬,并获得平均效果:

set.seed(2129)  (ex.jack <- explain(rfo, X = X, pred_wrapper = pfun, newdata = jack.dawson,                    nsim = 1000))##           pclass         age        sex       sibsp       parch## [1,] -0.07554003 -0.01240914 -0.1414107 0.001836116 -0.01103988## attr(,"baseline")## [1] 0## attr(,"class")## [1] "explain" "matrix"  "array"

fastshap使用的MC标准策画的Shapley值的加和不会就是相应预见和基线(即平均预见值)之间的差值。然则借用Python的shap库的手段,咱们不错使用基于转头的调整来更正总额。为此,只需在调用explain()中缔造为adjust = TRUE:

set.seed(2133)  (ex.jack.adj <- explain(rfo, X = X, pred_wrapper = pfun, newdata = jack.dawson,                        nsim = 1000, adjust = TRUE))##          pclass         age        sex       sibsp       parch## [1,] -0.0697378 -0.02354202 -0.1485205 0.003980237 -0.01281207## attr(,"baseline")## [1] 0.3821045## attr(,"class")## [1] "explain" "matrix"  "array"

这个效果的加和与difference是不同样的:

sum(ex.jack.adj)  # 和jack.prob - baseline的效果不同样哦## [1] -0.2506322

构建好讲明器之后,就不错使用shapviz包进行可视化了。

咱们先创建一个肤浅的瀑布图来可视化这几个特征是若何让Jack的预见生涯概率较低的:

library(shapviz)shv <- shapviz(ex.jack.adj, X = jack.dawson, baseline = baseline)sv_waterfall(shv)

图片

显着,Jack是又名男性三等舱乘客,这两个变量对他的较低生涯概率孝敬最大。

然后是瀑布图的另一种神态:

sv_force(shv)

图片

从图形清爽方面来看,这个图天然和瀑布图抒发的情理完全同样,然则不如瀑布图读起来肤浅。

全局讲明

SHAP除了用于局部讲明外,还不错用于全局讲明。

要是对数据兼并统统的不雅测齐进行一遍SHAP讲明,然后团员它们的效果,就不错得到全局的SHAP讲明。这个标准和CP团员成为PDP的标准绝顶访佛。

底下的代码使用1000次MC重迭策画检修数据中每个乘客的Shapley讲明,并将生成的矩阵强制调理为tibble(以便更好地打印)。缔造shap_only=FALSE不错便捷shapviz使用。

这个历程很慢哈~

set.seed(2224)  ex.t1 <- explain(rfo, X = X, pred_wrapper = pfun, nsim = 100, adjust = TRUE,                 shap_only = FALSE)tibble::as_tibble(ex.t1$shapley_values)## # A tibble: 1,309 × 5##    pclass      age     sex     sibsp    parch##     <dbl>    <dbl>   <dbl>     <dbl>    <dbl>##  1  0.228  0.00994  0.312   0.0208   -0.00916##  2  0.138  0.331   -0.0717  0.0113    0.0712 ##  3  0.156  0.0213   0.116  -0.0240   -0.0189 ##  4  0.212 -0.0212  -0.183   0.0171    0.00782##  5  0.201 -0.0308   0.282  -0.0149   -0.0369 ##  6  0.168 -0.0393  -0.199  -0.000779 -0.00240##  7  0.177 -0.127    0.346  -0.00829   0.00371##  8  0.153 -0.0644  -0.186  -0.00392  -0.00976##  9  0.234  0.00345  0.296   0.0398    0.00432## 10  0.106 -0.115   -0.212   0.000223 -0.00402## # ℹ 1,299 more rows

可视化全局变量关键性(防御和基于重排的变量关键性分辩),这个其实是各个变量的多个Shapley值完全值的平均值:

shv.global <- shapviz(ex.t1)sv_importance(shv)  

图片

变量关键性图的另一种阐述神态,蜂窝图,在Python中被称为shap summary plot:

sv_importance(shv.global, kind = "beeswarm")

图片

变量依赖图,访佛于部分依赖图。在这里,咱们将望望特征孝敬age对其输入值的依赖性:

sv_dependence(shv.global, v = "age")

图片

其他可视化函数可参考对于shapviz的推文:R话语shapviz可视化SHAP

要是你的数据量很大,fastshap还营救使用并行化色中色电影,借助doParallel包即可杀青,全球我方尝试下即可。

本站仅提供存储管事,统统骨子均由用户发布,如发现存害或侵权骨子,请点击举报。