FIX TSNE.fit: didn't return self

Author: larsmans

sklearn/manifold/t_sne.py

@@ -463,6 +463,8 @@ def fit(self, X, y=None):
         self.embedding_ = self._tsne(P, alpha, n_samples, random_state,
                                      X_embedded=X_embedded)
 
+        return self
+
     def _tsne(self, P, alpha, n_samples, random_state, X_embedded=None):
         """Runs t-SNE."""
         # t-SNE minimizes the Kullback-Leiber divergence of the Gaussians P

sklearn/manifold/tests/test_t_sne.py

@@ -269,5 +269,5 @@ def test_reduction_to_one_component():
     random_state = check_random_state(0)
     tsne = TSNE(n_components=1)
     X = random_state.randn(5, 2)
-    X_embedded = tsne.fit_transform(X)
+    X_embedded = tsne.fit(X).embedding_
     assert(np.all(np.isfinite(X_embedded)))