clustering.py 文件源码

def fitGMM(data, num_clusters, iterations, update, covar_type):
    print ('fit gmm with num_clusters {}, params {}, n_iter {}, covar_type '
        '{}'.format(num_clusters, update, iterations, covar_type))
    gmm = mixture.GMM(num_clusters, n_iter=iterations,
                      params=update, covariance_type=covar_type)
    gmm.fit(data)
    return gmm