FIX GMM 1d issue

jmschrei · Oct 9, 2017 · 7a352e2 · 7a352e2
1 parent 33865c4
commit 7a352e2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/docs/whats_new.rst b/docs/whats_new.rst
@@ -75,6 +75,9 @@ General Mixture Models
 	mean that out-of-core updates will have a different initialization now, but then
 	yield exact updates after that.
 
+	- Fixed bug where passing in a 1D array would cause an error by recasting all 1D
+	arrays as 2D arrays.
+
 Bayesian Networks
 .................
 

diff --git a/pomegranate/utils.pyx b/pomegranate/utils.pyx
@@ -323,6 +323,7 @@ def _check_input(X, keymap):
 			if X.ndim == 1:
 				for i in range(X.shape[0]):
 					X_ndarray[i] = keymap[0][X[i]]
+				X_ndarray = X_ndarray.reshape(-1, 1)
 			else:
 				for i in range(X.shape[0]):
 					for j in range(X.shape[1]):