1

AWS databricks GPU 实例似乎只提供 CPU 版本的 shap 库。按照 文档，我可以用 GPU 版本替换它（完成没有错误）。不幸的是，当在单个虚拟样本上使用它时，它会抛出无用的

ConnectException：连接被拒绝（连接被拒绝）获取新通信通道时出错

这应该是一个 OOM 错误 - 除了应该有足够的空间和神经节显示。我想我在这里陷入了死胡同。有什么方法可以获得更有用的错误信息？尝试其他婴儿步骤？还是可以完成相同事情的不同方法（在 databricks-AWS 中运行 GPUTree）？

4

1 回答 1

0

与树解释器一样，GPUTree 解释器专为基于树的机器学习模型而设计，但它旨在使用 NVIDIA GPU 加速计算。资源

要运行它，需要：

  1. shap安装在带有 GPU 的机器上的软件包
  2. 在运行时具有GPU 的机器。

否则，请尝试KernelExplainer，您可能会发现它有点慢

于 2022-01-27T03:28:10.623 回答