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

Implement clustering accuracy #2767

Open
moetayuko opened this issue Oct 3, 2024 · 1 comment
Open

Implement clustering accuracy #2767

moetayuko opened this issue Oct 3, 2024 · 1 comment
Labels
enhancement New feature or request

Comments

@moetayuko
Copy link

🚀 Feature

Motivation

Clustering accuracy is a popular metric. In addition to classification accuracy, it employs the Hungarian algorithm to align the predicted pseudo labels and the ground truth labels.

Current implementations of clustering accuracy use either scipy.optimize.linear_sum_assignment or the munkres package for Hungarian. I'm not sure if this is allowed for torchmetrics, and a custom implementation needs to be added if not.

Pitch

Implement clustering accuracy in torchmetrics.clustering

@moetayuko moetayuko added the enhancement New feature or request label Oct 3, 2024
Copy link

github-actions bot commented Oct 3, 2024

Hi! thanks for your contribution!, great first issue!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant