Kernel PCA: Python and Benchmarking Code #5988
Open
+902
−5
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Description
This PR relies on C++ implementation from #5987
Adds Python, and benchmarking code for Kernel PCA. This implementation of Kernel PCA support fit(), transform(), and fit_transform().
Feature request: #1317
Tests and benchmarks were performed on an EC2
g4dn.xlarge
instance with CUDA 12.2.Click here to see environment details
Notes for Reviewers
The API deviates from SKlearn by not supporting options for these fields: fit_inverse_transform, random_state, n_jobs, max_iter. If a user tries to set one of them a
NotImplementedError
will be raised.The Criteria of Done mentions making the class pickable in
cuml/tests/test_pickle.py
. I couldn't find a PCA reference for this. Would appreciate pointers if additional work is needed.Benchmarks
From
notebooks/tools/cuml_benchmarks.ipynb
Benchmark output
We see an even greater speedup when we set n_components = n_samples. Setting n_components to n_samples is the same as default behavior, except zero eigenvalues aren't removed.
Manual tests
Kernel PCA with RBF kernel
code
Kernel PCA with poly kernel
code
Projecting testing data
Case is copied from sklearn, except it uses cuML PCA and kernelPCA
code
Definition of Done Criteria Checklist
Python Checklist
Design
cuml/tests/test_pickle.py
input_to_cuml_array
to accept flexible inputs and check their datatypes and usecumlArray.to_output()
to return configurable outputs.CumlArray
Testing
python/cuml/benchmarks/algorithms.py
and benchmarks notebook inpython/cuml/notebooks/tools/cuml_benchmarks.ipynb
Unit test results
Python Test Results