diff options
| -rw-r--r-- | lib/functions.py | 2 |
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) |