diff options
| author | Daniel Friesel <daniel.friesel@uos.de> | 2022-06-15 08:34:49 +0200 |
|---|---|---|
| committer | Daniel Friesel <daniel.friesel@uos.de> | 2022-06-15 08:34:49 +0200 |
| commit | 68f0f50e62d801e1398c7a877e5ccaaa0d1b10ec (patch) | |
| tree | 6512a8469a3c4098b5bae7315497c675e43a7c28 /lib | |
| parent | 675a0f8538ee50542076818e2769c1d762fb42b5 (diff) | |
FOL, SKLearn: cast all parameters to int
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/functions.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/lib/functions.py b/lib/functions.py index 2b0b6e0..7cc8cdd 100644 --- a/lib/functions.py +++ b/lib/functions.py @@ -484,7 +484,7 @@ class SKLearnRegressionFunction(ModelFunction): max(self.categorial_to_index[i].values()) + 1 ) else: - actual_param_list.append(param) + actual_param_list.append(int(param)) predictions = self.regressor.predict(np.array([actual_param_list])) if predictions.shape == (1,): return predictions[0] @@ -714,7 +714,7 @@ class FOLFunction(ModelFunction): max(self.categorial_to_index[i].values()) + 1 ) else: - actual_param_list.append(param) + actual_param_list.append(int(param)) try: return self._function(self.model_args, actual_param_list) except FloatingPointError as e: |