diff options
| -rw-r--r-- | lib/parameters.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/lib/parameters.py b/lib/parameters.py index 4b6dc2c..401c7c6 100644 --- a/lib/parameters.py +++ b/lib/parameters.py @@ -924,6 +924,11 @@ class ModelAttribute: fit_parameters, category_to_index, ignore_index = param_to_ndarray( parameters, with_nan=False, categorial_to_scalar=categorial_to_scalar ) + if fit_parameters.shape[1] == 0: + logger.warning( + f"Cannot generate CART due to lack of parameters: parameter shape is {np.array(parameters).shape}, fit_parameter shape is {fit_parameters.shape}" + ) + return df.StaticFunction(np.mean(data)) cart.fit(fit_parameters, data) self.model_function = df.SKLearnRegressionFunction( np.mean(data), cart, category_to_index, ignore_index @@ -945,6 +950,11 @@ class ModelAttribute: fit_parameters, category_to_index, ignore_index = param_to_ndarray( parameters, with_nan=False, categorial_to_scalar=categorial_to_scalar ) + if fit_parameters.shape[1] == 0: + logger.warning( + f"Cannot run XGBoost due to lack of parameters: parameter shape is {np.array(parameters).shape}, fit_parameter shape is {fit_parameters.shape}" + ) + return df.StaticFunction(np.mean(data)) xgb.fit(fit_parameters, np.reshape(data, (-1, 1))) self.model_function = df.SKLearnRegressionFunction( np.mean(data), xgb, category_to_index, ignore_index |