diff --git a/bob/bio/gmm/algorithm/GMM.py b/bob/bio/gmm/algorithm/GMM.py index 8548a62ad2a0e65a6290e873a55ed919b2ff923f..3fb0d2c138b9f5bebb3dfd5965ed849d4803117e 100644 --- a/bob/bio/gmm/algorithm/GMM.py +++ b/bob/bio/gmm/algorithm/GMM.py @@ -302,6 +302,7 @@ class GMM(BioAlgorithm, BaseEstimator): init_method="k-means||", init_max_iter=self.kmeans_init_iterations, random_state=self.init_seed, + oversampling_factor=self.kmeans_oversampling_factor, ), ) diff --git a/bob/bio/gmm/test/test_gmm.py b/bob/bio/gmm/test/test_gmm.py index 7a7c454d28d412b2d3f73eaf57324992aed43ea3..809d8d3626eaf21870211c93b39b1a54c7f8c049 100644 --- a/bob/bio/gmm/test/test_gmm.py +++ b/bob/bio/gmm/test/test_gmm.py @@ -59,6 +59,7 @@ def test_training(): kmeans_training_iterations=5, ubm_training_iterations=5, init_seed=seed_value, + kmeans_oversampling_factor=2, ) train_data = utils.random_training_set( (100, 45), count=5, minimum=-5.0, maximum=5.0