From a589662fb2608d0317a363c78811e9c9f08e7525 Mon Sep 17 00:00:00 2001
From: Elie KHOURY <elie.khoury@idiap.ch>
Date: Wed, 24 Jun 2015 18:07:57 +0200
Subject: [PATCH] added the rng parameter in kmean initialization
---
bob/bio/gmm/tools/gmm.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/bob/bio/gmm/tools/gmm.py b/bob/bio/gmm/tools/gmm.py
index dd18cce..a35c94f 100644
--- a/bob/bio/gmm/tools/gmm.py
+++ b/bob/bio/gmm/tools/gmm.py
@@ -29,7 +29,7 @@ def kmeans_initialize(algorithm, extractor, limit_data = None, force = False):
# Perform KMeans initialization
kmeans_machine = bob.learn.em.KMeansMachine(algorithm.gaussians, data.shape[1])
# Creates the KMeansTrainer and call the initialization procedure
- algorithm.kmeans_trainer.initialize(kmeans_machine, data)
+ algorithm.kmeans_trainer.initialize(kmeans_machine, data, algorithm.rng)
bob.io.base.create_directories_safe(os.path.dirname(output_file))
kmeans_machine.save(bob.io.base.HDF5File(output_file, 'w'))
logger.info("UBM training: saved initial KMeans machine to '%s'", output_file)
--
GitLab