fix categorial-to-scalar handler for categorial values unseen during training

1 files changed, 1 insertions, 1 deletions

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)