From 60bcc3818281a388e7cab688c4c7de2ae38fc1a2 Mon Sep 17 00:00:00 2001
From: Yannick DAYER <yannick.dayer@idiap.ch>
Date: Fri, 4 Mar 2022 00:46:44 +0100
Subject: [PATCH] Fix the oversampling_factor not applied through.
---
bob/bio/gmm/algorithm/GMM.py | 1 +
bob/bio/gmm/test/test_gmm.py | 1 +
2 files changed, 2 insertions(+)
diff --git a/bob/bio/gmm/algorithm/GMM.py b/bob/bio/gmm/algorithm/GMM.py
index 8548a62..3fb0d2c 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 7a7c454..809d8d3 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
--
GitLab