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

[REVIEW] Adjusted Rand Index implementation #652

Conversation

venkywonka
Copy link
Contributor

  • CUDA implementation of the adjusted rand index metric
  • gtests for the same
  • wrapper for exposing to the cuML API

Ganesh Venkataramana added 5 commits June 4, 2019 03:16
- implemented CUDA header that calculates the adjusted rand index between two clusterings
- implemented the gtests for the same
- modifying the dOxygen comments
- minor changes to formatting and comments
- reduced a redundant parameter that the adjusted-rand-index function implementation takes as input
- wrote a wrapper to expose this module to cuml API
- Merged remote-tracking branch 'upstream/branch-0.8' into fea-ext-adjusted-rand-index-implementation
cpp/src/metrics/metrics.cu Outdated Show resolved Hide resolved
cpp/src/metrics/metrics.cu Outdated Show resolved Hide resolved
cpp/test/CMakeLists.txt Show resolved Hide resolved
@cjnolet cjnolet self-requested a review June 4, 2019 14:50
@cjnolet cjnolet changed the title Adjusted Rand Index implementation [WIP] Adjusted Rand Index implementation Jun 4, 2019
- changed the wrapper parameter types of y and y-hat from double to int since they represent nominal/ordinal data
- added `prims/add_sub_dev_scalar.cu` to CMakeLists.txt
Copy link
Member

@teju85 teju85 left a comment

Choose a reason for hiding this comment

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

Changes LGTM.
@dantegd @cjnolet please merge

@cjnolet
Copy link
Member

cjnolet commented Jun 7, 2019 via email

Ganesh Venkataramana added 3 commits June 6, 2019 22:16
- refactored the types starting from the wrapper function to it's traces in further functions
@teju85
Copy link
Member

teju85 commented Jun 13, 2019

Were you able to review this @cjnolet ?

Copy link
Member

@cjnolet cjnolet left a comment

Choose a reason for hiding this comment

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

One small change, otherwise this PR LGTM.


//workspace allocation
char* pWorkspace = nullptr;
size_t workspaceSz = MLCommon::Metrics::getCMatrixWorkspaceSize(
Copy link
Member

Choose a reason for hiding this comment

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

Let's use a more descriptive name since this function is sharing the namespace with everything else under metrics.

Copy link
Contributor Author

Choose a reason for hiding this comment

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

done!

@cjnolet cjnolet changed the title [WIP] Adjusted Rand Index implementation [REVIEW] Adjusted Rand Index implementation Jun 15, 2019
@cjnolet cjnolet added 3 - Ready for Review Ready for review by team CUDA / C++ CUDA issue New Prim For tracking new prims that will be added to our existing collection labels Jun 15, 2019
- changed the 'getCMatrixWorkspaceSize' method in contingencyMatrix.h and all it's other occurrances to 'getContingencyMatrixWorkspaceSize'.
Copy link
Member

@cjnolet cjnolet left a comment

Choose a reason for hiding this comment

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

Looks great. Will merge after conflicts are fixed.

@cjnolet cjnolet merged commit bc6cd6a into rapidsai:branch-0.8 Jun 18, 2019
@cjnolet
Copy link
Member

cjnolet commented Aug 21, 2019

#608

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
3 - Ready for Review Ready for review by team CUDA / C++ CUDA issue New Prim For tracking new prims that will be added to our existing collection
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants