Query - Is contextual-ai library shap and lime tested on GPU environment ?
manduripramodh opened this issue · 1 comments
manduripramodh commented
Hi ,
Is contextual-ai library shap and lime tested on GPU environment ? If yes, adding more GPU's will it improve the performance
wangjin1024 commented
Hi @manduripramodh, for now the shap and lime are mainly ran on CPU only as the library doesn't provide GPU implementation yet. Feel free to folk and contribute a GPU implementation or any parallel-based implementation to boost the performance.