Update t_sne.pyx for correctly transferring embedding_ to CPU

rapidsai · Dec 24, 2024 · 8058856 · 8058856
1 parent 12b2e3e
commit 8058856
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cuml/cuml/manifold/t_sne.pyx b/python/cuml/cuml/manifold/t_sne.pyx
@@ -727,4 +727,4 @@ class TSNE(UniversalBase,
     def get_attr_names(self):
         return ["embedding", "kl_divergence_",
                 "n_features_in_", "learning_rate_",
-                "n_iter_"]
+                "n_iter_", "embedding_"]