614 B
614 B
SHAP Tests
GPU TreeExplainer Tests
The GPU TreeExplainer tests require a CUDA-capable GPU and the _cext_gpu extension
built from source. Since GPU hardware is not available in our CI pipeline, these tests
can be run manually on Google Colab.
Running on Google Colab
- Click the badge above to open the notebook in Google Colab
- Switch to a GPU runtime: Runtime > Change runtime type > T4 GPU
- Run all cells