shap不能做多变量回归？为什么老爆内存？

接上文，在解决了catboost可以做多任务回归之后，shap又遇到了kernel崩溃的问题，心态崩溃ing，如下图。既然是深度太深了爆内存，那么就把catboost的树的深度降到最低就好了，我这里试了试大概是5层。在尝试了各种解决方法之后，发现是因为SHAP跑的时候，树的深度太深会使得内存爆掉。吐槽一下SHAP网上资料太少了。

victor_manches

437人浏览 · 2024-07-16 23:43:49

victor_manches · 2024-07-16 23:43:49 发布

问题

接上文，在解决了catboost可以做多任务回归之后，shap又遇到了kernel崩溃的问题，心态崩溃ing，如下图
在这里插入图片描述

原因

在尝试了各种解决方法之后，发现是因为SHAP跑的时候，树的深度太深会使得内存爆掉。

解决方法

既然是深度太深了爆内存，那么就把catboost的树的深度降到最低就好了，我这里试了试大概是5层。问题完美解决！
吐槽一下SHAP网上资料太少了。。。

在这里插入图片描述

2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

UFW防火墙安全指南

UFW（Uncomplicated Firewall）是Ubuntu/Debian系统中简化防火墙管理的工具，通过直观命令帮助用户有效控制网络流量，提升系统安全性。文章详细介绍了UFW的基本命令，包括启停防火墙、添加规则、限制连接速率和日志配置等操作，并提供了安全最佳实践，如默认拒绝策略、IP地址限制和服务级规则管理。同时，还涵盖高级配置技巧，例如多网络接口设置、规则优先级调整、IPv6支持及与f