def gmm_learn(v, k, nt=1, niter=30, seed=0, redo=1, use_weights=True):
_check_row_float32(v)
n, d = v.shape
flags = 0
if use_weights: flags |= yael.GMM_FLAGS_W
gmm = yael.gmm_learn(d, n, k, niter, fvec_ref(v), nt, seed, redo, flags)
gmm_npy = _gmm_to_numpy(gmm)
yael.gmm_delete(gmm)
return gmm_npy
def gmm_learn(v, k, nt=1, niter=30, seed=0, redo=1, use_weights=True):
_check_row_float32(v)
n, d = v.shape
flags = 0
if use_weights:
flags |= yael.GMM_FLAGS_W
gmm = yael.gmm_learn(d, n, k, niter, yael.numpy_to_fvec_ref(v), nt, seed, redo, flags)
gmm_npy = _gmm_to_numpy(gmm)
yael.gmm_delete(gmm)
return gmm_npy
def gmm_read(filename):
gmm = yael.gmm_read(open(filename, "r"))
gmm_npy = _gmm_to_numpy(gmm)
yael.gmm_delete(gmm)
return gmm_npy
def _gmm_del(gmm):
gmm.mu = gmm.sigma = gmm.w = None
yael.gmm_delete(gmm)
