如何在AWS数据库上使用GPUTree ( shap解释程序)？

Question

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

ConnectException:当获得新的通信信道时，连接拒绝(连接拒绝)错误

这被认为是OOM错误--此外，应该有足够的空间，而ganglia显示了这一点。我想我已经到了死胡同了。有没有办法得到更有用的错误信息？其他的婴儿步骤可以试试吗？或者用不同的方法来完成同样的事情(在databricks中运行GPUTree )？

Answer 1

和Tree一样，GPUTree解释器是专门为基于树的机器学习模型设计的，但它的设计目的是使用NVIDIA加速计算。来源

要运行它，需要：

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

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