Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
aj280192 authored Mar 29, 2022
1 parent 0ff8514 commit a69562d
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ arXiv pre-print available here: https://arxiv.org/abs/2108.13961

### With pip

[![PyPI](https://img.shields.io/pypi/v/thermostat-datasets?style=flat-square)](https://pypi.org/project/thermostat-datasets/)
[![PyPI](https://img.shields.io/pypi/v/thermostat-datasets?style=flat-square)](https://pypi.org/project/thermostat-datasets/)]

```bash
pip install thermostat-datasets
Expand Down Expand Up @@ -144,8 +144,8 @@ Layer Integrated Gradients (`lig`) | [`.attr.LayerIntegratedGradients`](https://
LIME (`lime`) | [`.attr.LimeBase`](https://captum.ai/api/lime.html) | # samples = 25, <br>mask prob = 0.3
Occlusion (`occ`) | [`.attr.Occlusion`](https://captum.ai/api/occlusion.html) | sliding window = 3
Shapley Value Sampling (`svs`) | [`.attr.ShapleyValueSampling`](https://captum.ai/api/shapley_value_sampling.html) | # samples = 25
Layer DeepLiftShap ('lds') | ['.attr.LayerDeepLiftShap'](https://captum.ai/api/layer.html#layer-deepliftshap) |
Layer GradientShap ('lgs') | ['attr.LayerGradientShap'](https://captum.ai/api/layer.html#layer-gradientshap) | # samples = 5
Layer DeepLiftShap ('lds') | [`.attr.LayerDeepLiftShap`](https://captum.ai/api/layer.html#layer-deepliftshap) |
Layer GradientShap ('lgs') | [`attr.LayerGradientShap`](https://captum.ai/api/layer.html#layer-gradientshap) | # samples = 5



Expand All @@ -165,7 +165,7 @@ Layer GradientShap ('lgs') | ['attr.LayerGradientShap'](https://captum.ai/api/la
Example configuration: `imdb-xlnet-lig`

Name | 🤗 | `lgxa` | `lig` | `lime` | `occ` | `svs` | 'lds' | 'lgs'
--- | --- | --- | --- | --- | --- | ---
--- | --- | --- | --- | --- | --- | --- | --- | ---
ALBERT (`albert`) | [`textattack/albert-base-v2-imdb`](https://huggingface.co/textattack/albert-base-v2-imdb) | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅
BERT (`bert`) | [`textattack/bert-base-uncased-imdb`](https://huggingface.co/textattack/bert-base-uncased-imdb) | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅
ELECTRA (`electra`) | [`monologg/electra-small-finetuned-imdb`](https://huggingface.co/monologg/electra-small-finetuned-imdb) | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅
Expand All @@ -179,7 +179,7 @@ XLNet (`xlnet`) | [`textattack/xlnet-base-cased-imdb`](https://huggingface.co/te
Example configuration: `multi_nli-roberta-lime`

Name | 🤗 | `lgxa` | `lig` | `lime` | `occ` | `svs` | 'lds' | 'lgs'
--- | --- | --- | --- | --- | --- | ---
--- | --- | --- | --- | --- | --- | --- | --- | ---
ALBERT (`albert`) | [`prajjwal1/albert-base-v2-mnli`](https://huggingface.co/prajjwal1/albert-base-v2-mnli) | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅
BERT (`bert`) | [`textattack/bert-base-uncased-MNLI`](https://huggingface.co/textattack/bert-base-uncased-MNLI) | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅
ELECTRA (`electra`) | [`howey/electra-base-mnli`](https://huggingface.co/howey/electra-base-mnli) | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅
Expand All @@ -193,7 +193,7 @@ XLNet (`xlnet`) | [`textattack/xlnet-base-cased-MNLI`](https://huggingface.co/te
Example configuration: `xnli-roberta-lime`

Name | 🤗 | `lgxa` | `lig` | `lime` | `occ` | `svs` | 'lds' | 'lgs'
--- | --- | --- | --- | --- | --- | ---
--- | --- | --- | --- | --- | --- | --- | --- | ---
ALBERT (`albert`) | [`prajjwal1/albert-base-v2-mnli`](https://huggingface.co/prajjwal1/albert-base-v2-mnli) | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅
BERT (`bert`) | [`textattack/bert-base-uncased-MNLI`](https://huggingface.co/textattack/bert-base-uncased-MNLI) | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅
ELECTRA (`electra`) | [`howey/electra-base-mnli`](https://huggingface.co/howey/electra-base-mnli) | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅
Expand All @@ -207,7 +207,7 @@ XLNet (`xlnet`) | [`textattack/xlnet-base-cased-MNLI`](https://huggingface.co/te
Example configuration: `ag_news-albert-svs`

Name | 🤗 | `lgxa` | `lig` | `lime` | `occ` | `svs` | 'lds' | 'lgs'
--- | --- | --- | --- | --- | --- | ---
--- | --- | --- | --- | --- | --- | --- | --- | ---
ALBERT (`albert`) | [`textattack/albert-base-v2-ag-news`](https://huggingface.co/textattack/albert-base-v2-ag-news) | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅
BERT (`bert`) | [`textattack/bert-base-uncased-ag-news`](https://huggingface.co/textattack/bert-base-uncased-ag-news) | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅
RoBERTa (`roberta`) | [`textattack/roberta-base-ag-news`](https://huggingface.co/textattack/roberta-base-ag-news) | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅
Expand Down

0 comments on commit a69562d

Please sign in to comment.