Merge pull request rapidsai#4387 from rapidsai/branch-21.12

[gpuCI] Forward-merge branch-21.12 to branch-22.02 [skip gpuci]
vimarsh6739 · Nov 18, 2021 · 2edcbb8 · 2edcbb8
2 parents d498153 + b48c4f7
commit 2edcbb8
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/python/cuml/manifold/t_sne.pyx b/python/cuml/manifold/t_sne.pyx
@@ -172,7 +172,8 @@ class TSNE(Base,
         0.8. (Barnes-Hut only.)
     learning_rate_method : str 'adaptive', 'none' or None (default 'adaptive')
         Either adaptive or None. 'adaptive' tunes the learning rate, early
-        exaggeration and perplexity automatically based on input size.
+        exaggeration, perplexity and n_neighbors automatically based on
+        input size.
     n_neighbors : int (default 90)
         The number of datapoints you want to use in the
         attractive forces. Smaller values are better for preserving
@@ -482,11 +483,6 @@ class TSNE(Base,
         if self.method == 'barnes_hut':
             algo = TSNE_ALGORITHM.BARNES_HUT
         elif self.method == 'fft':
-            warnings.warn("Method 'fft' is experimental and may be " +
-                          "unstable. If you find this implementation is not" +
-                          " behaving as intended, please consider using one" +
-                          " of the other methods, such as 'barnes_hut' or" +
-                          " 'exact'")
             algo = TSNE_ALGORITHM.FFT
         elif self.method == 'exact':
             algo = TSNE_ALGORITHM.EXACT
@@ -529,8 +525,7 @@ class TSNE(Base,
         free(params)
 
         self._kl_divergence_ = kl_divergence
-        if self.verbose:
-            print("[t-SNE] KL divergence: {}".format(kl_divergence))
+        logger.debug("[t-SNE] KL divergence: {}".format(kl_divergence))
         return self
 
     @generate_docstring(convert_dtype_cast='np.float32',