From b55023fda36ec320be4f7cd41ec83a9914a5d165 Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Wed, 21 Feb 2024 12:40:57 +0100 Subject: build_dtree: there is no need to explicitly pass param_values and data --- lib/model.py | 4 ---- lib/parameters.py | 6 ++---- 2 files changed, 2 insertions(+), 8 deletions(-) (limited to 'lib') 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, -- cgit v1.2.3