fix: multi dimension update for covariance in gmm

WuZhuoran · WuZhuoran · commit bea2967f514c · 2019-07-11T10:37:52.000-07:00
diff --git a/gmm/gmm.py b/gmm/gmm.py
@@ -120,7 +120,7 @@ def _M_step(self):
             mu_c = self.mu[c, :]
             n_c = denoms[c]
 
-            outer = np.zeros((2, 2))
+            outer = np.zeros((self.d, self.d))
             for i in range(N):
                 wic = self.Q[i, c]
                 xi = self.X[i, :]
