From 805885693a67663e3f2072ebc4dd6fd22fb8cab2 Mon Sep 17 00:00:00 2001 From: Dante Gama Dessavre Date: Tue, 24 Dec 2024 10:14:40 -0500 Subject: [PATCH] Update t_sne.pyx for correctly transferring embedding_ to CPU --- python/cuml/cuml/manifold/t_sne.pyx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/cuml/cuml/manifold/t_sne.pyx b/python/cuml/cuml/manifold/t_sne.pyx index 01ea7c0957..0d047c59d0 100644 --- 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_"]