From f602806133ea712a7a353ca75702346ec5288b27 Mon Sep 17 00:00:00 2001 From: abigailt Date: Mon, 4 Dec 2023 08:52:15 -0500 Subject: [PATCH] Indication when fitting failed Signed-off-by: abigailt --- apt/minimization/minimizer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/apt/minimization/minimizer.py b/apt/minimization/minimizer.py index d91bd5a..39521d4 100644 --- a/apt/minimization/minimizer.py +++ b/apt/minimization/minimizer.py @@ -396,6 +396,8 @@ class GeneralizeToRepresentative(BaseEstimator, MetaEstimatorMixin, TransformerM x_test_dataset = ArrayDataset(x_test, features_names=self._features) self._ncp_scores.fit_score = self.calculate_ncp(x_test_dataset) self._ncp_scores.generalizations_score = self.calculate_ncp(x_test_dataset) + else: + print('No fitting was performed as some information was missing') # Return the transformer return self