use our GMM.

bccp · Sep 28, 2016 · 7a79412 · 7a79412
1 parent 5191a30
commit 7a79412
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/bananas/bananas.py b/bananas/bananas.py
@@ -217,7 +217,7 @@ def compile(self, features, nc=1, nb=20, cov="full", ):
 
         data = numpy.concatenate(data, axis=0)
 
-        from sklearn import mixture
+        from . import gmm as mixture
 
         # XXX: Do not use DPGMM because the normalization is buggy
         # https://github.com/scikit-learn/scikit-learn/issues/7371

diff --git a/bananas/gmm.py b/bananas/gmm.py
@@ -14,11 +14,11 @@
 from scipy import linalg
 from time import time
 
-from ..base import BaseEstimator
-from ..utils import check_random_state, check_array
-from ..utils.extmath import logsumexp
-from ..utils.validation import check_is_fitted
-from .. import cluster
+from sklearn.base import BaseEstimator
+from sklearn.utils import check_random_state, check_array
+from sklearn.utils.extmath import logsumexp
+from sklearn.utils.validation import check_is_fitted
+from sklearn import cluster
 
 from sklearn.externals.six.moves import zip