From d39f402184ede7702b2823c60cd0066f4b1db310 Mon Sep 17 00:00:00 2001
From: Daniel Friesel <daniel.friesel@uos.de>
Date: Fri, 14 Jan 2022 16:27:40 +0100
Subject: fix categorial-to-scalar handler for categorial values unseen during
training
---
lib/functions.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
(limited to 'lib')
diff --git a/lib/functions.py b/lib/functions.py
index af83146..9bbc0e7 100644
--- a/lib/functions.py
+++ b/lib/functions.py
@@ -451,7 +451,7 @@ class SKLearnRegressionFunction(ModelFunction):
# Note that all param values which were not part of training data map to the same scalar this way.
# This should be harmless.
actual_param_list.append(
- max(self.categorial_to_index[i][param]) + 1
+ max(self.categorial_to_index[i].values()) + 1
)
else:
actual_param_list.append(param)
--
cgit v1.2.3