Compatibility python 3

unbalanced_dataset/pipeline.py

@@ -13,6 +13,8 @@
 #         Lars Buitinck
 #         chkoar
 # License: BSD
+from __future__ import print_function
+from __future__ import division
 
 from warnings import warn
 
@@ -373,9 +375,7 @@ def score(self, X, y=None):
             if hasattr(transform, "fit_sample"):
                 pass
             else:
-                print Xt.shape
                 Xt = transform.transform(Xt)
-                print Xt.shape
         return self.steps[-1][-1].score(Xt, y)
 
 
@@ -386,15 +386,6 @@ def make_pipeline(*steps):
     does not permit, naming the estimators. Instead, their names will be set
     to the lowercase of their types automatically.
 
-    Examples
-    --------
-    >>> from sklearn.naive_bayes import GaussianNB
-    >>> from sklearn.preprocessing import StandardScaler
-    >>> make_pipeline(StandardScaler(), GaussianNB(priors=None))    # doctest: +NORMALIZE_WHITESPACE
-    Pipeline(steps=[('standardscaler',
-                     StandardScaler(copy=True, with_mean=True, with_std=True)),
-                    ('gaussiannb', GaussianNB(priors=None))])
-
     Returns
     -------
     p : Pipeline