diff options
Diffstat (limited to 'lib/functions.py')
| -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 5358d8e..f4e1709 100644 --- a/lib/functions.py +++ b/lib/functions.py @@ -443,7 +443,7 @@ class SKLearnRegressionFunction(ModelFunction): for i, param in enumerate(param_list): if not self.ignore_index[i]: actual_param_list.append(param) - return self.regressor.predict([actual_param_list]) + return self.regressor.predict(np.array([actual_param_list])) class AnalyticFunction(ModelFunction): |