Skip to content

Add Wilcoxon's signed rank test#237

Merged
douwekiela merged 3 commits into
mainfrom
wilcoxon
Aug 11, 2022
Merged

Add Wilcoxon's signed rank test#237
douwekiela merged 3 commits into
mainfrom
wilcoxon

Conversation

@douwekiela

Copy link
Copy Markdown
Contributor

Figured it'd be good to add a few more comparisons

@douwekiela
douwekiela requested a review from lvwerra August 9, 2022 06:27
Comment thread comparisons/wilcoxon/README.md Outdated

```python
wilcoxon = evaluate.load("wilcoxon")
results = wilcoxon.compute(predictions1=[-7, 123, 43, 4, 5], predictions2=[1337, -9, 1, 2, 3])

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hrmm should probably make some of these floats to make clear that those are allowed too

@HuggingFaceDocBuilderDev

HuggingFaceDocBuilderDev commented Aug 9, 2022

Copy link
Copy Markdown

The documentation is not available anymore as the PR was closed or merged.

@lvwerra lvwerra left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Hi @douwekiela, thanks for the super clean PR. Left a comment regarding the feature types related to your own comment.

Comment on lines +64 to +69
features=datasets.Features(
{
"predictions1": datasets.Value("int64"),
"predictions2": datasets.Value("int64"),
}
),

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

If you want it to work for both int and float you can pass a list of dataset.Features and it then automatically detects which one works. You can have a look at BLEU. Alternatively, I think floats would probably work both cases anyway, no?

@douwekiela
douwekiela merged commit 3cd38e2 into main Aug 11, 2022
@douwekiela
douwekiela deleted the wilcoxon branch August 11, 2022 13:01
mathemakitten pushed a commit that referenced this pull request Aug 15, 2022
Add Wilcoxon's signed rank test for comparing model predictions, e.g. for testing whether the difference in BLEU score between two models is significant.
mathemakitten pushed a commit that referenced this pull request Sep 23, 2022
Add Wilcoxon's signed rank test for comparing model predictions, e.g. for testing whether the difference in BLEU score between two models is significant.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants