diff options
| author | Birte Kristina Friesel <birte.friesel@uos.de> | 2024-02-21 12:40:57 +0100 |
|---|---|---|
| committer | Birte Kristina Friesel <birte.friesel@uos.de> | 2024-02-21 12:40:57 +0100 |
| commit | b55023fda36ec320be4f7cd41ec83a9914a5d165 (patch) | |
| tree | 9b84d3810135e345ade019ece6f5a13db626459c | |
| parent | d5950d6f31de5403ed61124d799cd0fafe491b06 (diff) | |
build_dtree: there is no need to explicitly pass param_values and data
| -rw-r--r-- | lib/model.py | 4 | ||||
| -rw-r--r-- | lib/parameters.py | 6 |
2 files changed, 2 insertions, 8 deletions
diff --git a/lib/model.py b/lib/model.py index 6718090..715ccb1 100644 --- a/lib/model.py +++ b/lib/model.py @@ -327,8 +327,6 @@ class AnalyticModel: threshold = 0 logger.debug(f"build_dtree({name}, {attr}, threshold={threshold})") self.attr_by_name[name][attr].build_dtree( - self.by_name[name]["param"], - self.by_name[name][attr], threshold=threshold, ) else: @@ -374,8 +372,6 @@ class AnalyticModel: f"build_dtree({name}, {attr}, threshold={threshold})" ) self.attr_by_name[name][attr].build_dtree( - self.by_name[name]["param"], - self.by_name[name][attr], threshold=threshold, ) else: diff --git a/lib/parameters.py b/lib/parameters.py index d3d2659..e4499ce 100644 --- a/lib/parameters.py +++ b/lib/parameters.py @@ -1006,8 +1006,6 @@ class ModelAttribute: def build_dtree( self, - parameters, - data, with_function_leaves=None, with_nonbinary_nodes=None, with_gplearn_symreg=None, @@ -1059,8 +1057,8 @@ class ModelAttribute: ) self.model_function = self._build_dtree( - parameters, - data, + self.param_values, + self.data, with_function_leaves=with_function_leaves, with_nonbinary_nodes=with_nonbinary_nodes, ignore_irrelevant_parameters=ignore_irrelevant_parameters, |