Query - Is contextual-ai library shap and lime tested on GPU environment ?

SAP-archive / contextual-ai

Contextual AI adds explainability to different stages of machine learning pipelines - data, training, and inference - thereby addressing the trust gap between such ML systems and their users. It does not refer to a specific algorithm or ML method — instead, it takes a human-centric view and approach to AI.

https://contextual-ai.readthedocs.io/en/latest

Apache License 2.0

86 stars 12 forks source link

Query - Is contextual-ai library shap and lime tested on GPU environment ? #13

Closed manduripramodh closed 3 years ago

manduripramodh commented 4 years ago

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 3 years ago

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.