Bugs fixes and new benchmark code

surendramph · Aug 24, 2021 · 9d7de4d · 9d7de4d
1 parent 89855e0
commit 9d7de4d
Show file tree

Hide file tree

Showing 37 changed files with 3,966 additions and 636 deletions.
diff --git a/.pylintrc b/.pylintrc
@@ -60,7 +60,8 @@ confidence=
 # --enable=similarities". If you want to run only the classes checker, but have
 # no Warning level messages displayed, use "--disable=all --enable=classes
 # --disable=W".
-disable=print-statement,
+disable=unbalanced-tuple-unpacking,
+        print-statement,
         parameter-unpacking,
         unpacking-in-except,
         old-raise-syntax,

diff --git a/.travis.yml b/.travis.yml
@@ -20,7 +20,7 @@ install:
 - pip install tensorflow xgboost matplotlib ipython codecov torch torchvision pylint sentencepiece
 script:
   - python -m pytest tests
-  - pylint tests shap/models shap/maskers shap/explainers/_gpu_tree.py
+  - pylint tests shap/models shap/maskers shap/explainers/_gpu_tree.py shap/_serializable.py shap/links.py shap/datasets.py 
 after_success:
   - codecov
 deploy:

diff --git a/docs/benchmarks.rst b/docs/benchmarks.rst
@@ -5,8 +5,22 @@ These benchmark notebooks compare different types of explainers across a variety
 They are all generated from Jupyter notebooks
 `available on GitHub <https://github.com/slundberg/shap/tree/master/notebooks/benchmarks>`_.
 
+Tabular
+====================
+Benchmarks that compare explainers on tabular datasets.
+
+.. toctree::
+    :glob:
+    :maxdepth: 1
+
+    example_notebooks/benchmarks/tabular/*
+
+Text
+====================
+Benchmarks that compare explainers on text datasets.
+
 .. toctree::
     :glob:
     :maxdepth: 1
 
-    example_notebooks/benchmark/*
+    example_notebooks/benchmarks/text/*
diff --git a/notebooks/api_examples/explainers/GPUTree.ipynb b/notebooks/api_examples/explainers/GPUTree.ipynb
diff --git a/notebooks/benchmark/others/Explainers Benchmark Comparison.ipynb b/notebooks/benchmark/others/Explainers Benchmark Comparison.ipynb
diff --git a/notebooks/benchmarks/tabular/Benchmark XGBoost explanations.ipynb b/notebooks/benchmarks/tabular/Benchmark XGBoost explanations.ipynb
diff --git a/.../image_examples/image_classification/Explain ResNet50 using the Partition explainer.ipynb b/.../image_examples/image_classification/Explain ResNet50 using the Partition explainer.ipynb
diff --git a/notebooks/image_examples/image_classification/Image Multi Class.ipynb b/notebooks/image_examples/image_classification/Image Multi Class.ipynb
diff --git a/shap/.pylintrc2 b/shap/.pylintrc2
@@ -0,0 +1 @@
+generated-members=Explanation.*