update

Author: User

hmm_class/hmmc_theano.py

@@ -28,8 +28,7 @@ def __init__(self, M, K):
         self.K = K # number of Gaussians
 
     def fit(self, X, learning_rate=1e-2, max_iter=10):
-        # train the HMM model using the Baum-Welch algorithm
-        # a specific instance of the expectation-maximization algorithm
+        # train the HMM model using gradient descent
 
         N = len(X)
         D = X[0].shape[1] # assume each x is organized (T, D)

hmm_class/hmmc_theano2.py

@@ -30,8 +30,7 @@ def __init__(self, M, K):
         self.K = K # number of Gaussians
 
     def fit(self, X, learning_rate=1e-2, max_iter=10):
-        # train the HMM model using the Baum-Welch algorithm
-        # a specific instance of the expectation-maximization algorithm
+        # train the HMM model using gradient descent
 
         N = len(X)
         D = X[0].shape[1] # assume each x is organized (T, D)