From 28755e48ed0825e5c938db4a06012125cda026f9 Mon Sep 17 00:00:00 2001
From: Amir MOHAMMADI <amir.mohammadi@idiap.ch>
Date: Mon, 25 Apr 2022 18:22:11 +0200
Subject: [PATCH] Check for consistent length between X and y
---
bob/learn/em/factor_analysis.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/bob/learn/em/factor_analysis.py b/bob/learn/em/factor_analysis.py
index b65da82..ea66294 100644
--- a/bob/learn/em/factor_analysis.py
+++ b/bob/learn/em/factor_analysis.py
@@ -12,6 +12,7 @@ import numpy as np
from dask.array.core import Array
from dask.delayed import Delayed
from sklearn.base import BaseEstimator
+from sklearn.utils import check_consistent_length
from sklearn.utils.multiclass import unique_labels
from .gmm import GMMMachine
@@ -1260,8 +1261,10 @@ class FactorAnalysisBase(BaseEstimator):
return self.score_using_stats(model, self.ubm.transform(data))
def fit(self, X, y):
+
input_is_dask, X = check_and_persist_dask_input(X, persist=False)
y = np.squeeze(np.asarray(y))
+ check_consistent_length(X, y)
self.initialize(X)
--
GitLab