Skip to content
Snippets Groups Projects
Commit a589662f authored by Elie KHOURY's avatar Elie KHOURY
Browse files

added the rng parameter in kmean initialization

parent 88eca591
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
bob/bio/gmm/tools/gmm.py
+ 1
− 1
View file @ a589662f
...@@ -29,7 +29,7 @@ def kmeans_initialize(algorithm, extractor, limit_data = None, force = False): ...@@ -29,7 +29,7 @@ def kmeans_initialize(algorithm, extractor, limit_data = None, force = False):
# Perform KMeans initialization # Perform KMeans initialization
kmeans_machine = bob.learn.em.KMeansMachine(algorithm.gaussians, data.shape[1]) kmeans_machine = bob.learn.em.KMeansMachine(algorithm.gaussians, data.shape[1])
# Creates the KMeansTrainer and call the initialization procedure # 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)) bob.io.base.create_directories_safe(os.path.dirname(output_file))
kmeans_machine.save(bob.io.base.HDF5File(output_file, 'w')) kmeans_machine.save(bob.io.base.HDF5File(output_file, 'w'))
logger.info("UBM training: saved initial KMeans machine to '%s'", output_file) logger.info("UBM training: saved initial KMeans machine to '%s'", output_file)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment