Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

⚖️🌡️ Weighted Rank-Based Metrics #837

Merged
merged 80 commits into from
Mar 29, 2022
Merged

⚖️🌡️ Weighted Rank-Based Metrics #837

merged 80 commits into from
Mar 29, 2022

Conversation

mberr
Copy link
Member

@mberr mberr commented Mar 14, 2022

This PR adds support for individual weights for each ranking tasks, in preparation for adding macro evaluation.

It partially re-implements the changes #812 based upon the changes from #786.

In particular, it adds weighting options to

  • arithmetic mean rank
  • geometric mean rank
  • harmonic mean rank
  • median rank
  • median absolute deviation
  • hits at k

as well as their

  • inverses
  • expected values
    • AMR
    • GMR
    • IHMR
    • H@k
  • variances
    • AMR
    • GMR
    • IHMR
    • H@k
  • expectation normalized metrics
  • re-indexed metrics
  • z-adjusted metrics

Not solved by this PR and postponed to a separate follow-up PR:

  • calculating closed form expectation / variance of weighted GMR
  • calculating the macro-weights & adding an appropriate evaluator

@mberr mberr changed the title Macro Evaluation Weighted Rank-Based Metrics Mar 27, 2022
@mberr mberr requested a review from cthoyt March 28, 2022 16:13
@mberr mberr mentioned this pull request Mar 28, 2022
1 task
src/pykeen/cli.py Outdated Show resolved Hide resolved
README.md Outdated Show resolved Hide resolved
@mberr mberr merged commit 6ac40cd into master Mar 29, 2022
@mberr mberr deleted the macro-metrics-2 branch March 29, 2022 14:27
@mberr mberr mentioned this pull request Jun 9, 2022
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.

None yet

2 participants