-
Notifications
You must be signed in to change notification settings - Fork 540
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
Condense TSNE parameters into a struct #3884
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just a tiny suggestion / nitpick but otherwise PR looks great. Nice to see how much this cleaned up the arguments list.
python/cuml/manifold/t_sne.pyx
Outdated
@@ -509,6 +492,30 @@ class TSNE(Base, | |||
raise ValueError("Allowed methods are 'exact', 'barnes_hut' and " | |||
"'fft'.") | |||
|
|||
cdef TSNEParams params |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just a tiny nitpick- it might clean this method up even further if the TSNE class had a method like _build_tsne_params
that could run this logic.
Codecov Report
@@ Coverage Diff @@
## branch-21.06 #3884 +/- ##
===============================================
Coverage ? 85.40%
===============================================
Files ? 227
Lines ? 17297
Branches ? 0
===============================================
Hits ? 14772
Misses ? 2525
Partials ? 0
Flags with carried forward coverage won't be shown. Click here to find out more. Continue to review full report at Codecov.
|
@gpucibot merge |
Closes rapidsai#3781 Authors: - Micka (https://github.com/lowener) Approvers: - Corey J. Nolet (https://github.com/cjnolet) URL: rapidsai#3884
Closes #3781