From 7bf011912c44af3d3b220d70b352d183d0b91f49 Mon Sep 17 00:00:00 2001 From: Josep Maria Salvia Hornos Date: Tue, 14 Nov 2023 09:54:14 +0100 Subject: [PATCH] Ignore on onehotencoder fixes synthetizer unknown category errors --- sdmetrics/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdmetrics/utils.py b/sdmetrics/utils.py index 30e56ebe..b50713ad 100644 --- a/sdmetrics/utils.py +++ b/sdmetrics/utils.py @@ -158,7 +158,7 @@ def fit(self, data): elif kind == 'O': # Categorical column. col_data = pd.DataFrame({'field': data[field]}) - enc = OneHotEncoder() + enc = OneHotEncoder(handle_unknown='ignore') enc.fit(col_data) self.column_transforms[field] = {'one_hot_encoder': enc} elif kind == 'M':