Add IncrementalPCA to experimental CLI feature

rapidsai · Jan 24, 2025 · 078cebc · 078cebc
1 parent 0d9d09d
commit 078cebc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/python/cuml/cuml/decomposition/incremental_pca.py b/python/cuml/cuml/decomposition/incremental_pca.py
@@ -201,6 +201,7 @@ class IncrementalPCA(PCA):
     """
 
     _cpu_estimator_import_path = "sklearn.decomposition.IncrementalPCA"
+    _hyperparam_interop_translator = {}
 
     @device_interop_preparation
     def __init__(

diff --git a/python/cuml/cuml/experimental/accel/_wrappers/sklearn.py b/python/cuml/cuml/experimental/accel/_wrappers/sklearn.py
@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2024, NVIDIA CORPORATION.
+# Copyright (c) 2024-2025, NVIDIA CORPORATION.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -46,6 +46,13 @@
 )
 
 
+IncrementalPCA = intercept(
+    original_module="sklearn.decomposition",
+    accelerated_module="cuml.decomposition",
+    original_class_name="IncrementalPCA",
+)
+
+
 TruncatedSVD = intercept(
     original_module="sklearn.decomposition",
     accelerated_module="cuml.decomposition",