diff options
Diffstat (limited to 'lib/parameters.py')
| -rw-r--r-- | lib/parameters.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/lib/parameters.py b/lib/parameters.py index ad23780..ae4fffb 100644 --- a/lib/parameters.py +++ b/lib/parameters.py @@ -1062,6 +1062,7 @@ class ModelAttribute: category_to_index, ignore_index, n_samples=len(data), + param_names=self.param_names, ) logger.debug("Fitted sklearn CART") return |