diff --git a/bob/bio/gmm/tools/gmm.py b/bob/bio/gmm/tools/gmm.py index dd18ccedc08660b31bbcc902c2d1350ada7b9895..a35c94f2db6d112f4cd180af75ba2a8bfdccbe2e 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)