diff options
author | Birte Kristina Friesel <birte.friesel@uos.de> | 2024-02-21 13:10:52 +0100 |
---|---|---|
committer | Birte Kristina Friesel <birte.friesel@uos.de> | 2024-02-21 13:10:52 +0100 |
commit | 761df90cebc0e0fb443c1333f7d97347cd5ebd98 (patch) | |
tree | ce1a13465ac8c37f15d9f3b28e4200ed976e233c | |
parent | e525d288a3ce1d42462bed42d30037ad06c6e4ff (diff) |
rename build_dtree to build_rmt
-rw-r--r-- | lib/model.py | 8 | ||||
-rw-r--r-- | lib/parameters.py | 12 |
2 files changed, 10 insertions, 10 deletions
diff --git a/lib/model.py b/lib/model.py index 3d3e45d..26c4646 100644 --- a/lib/model.py +++ b/lib/model.py @@ -325,8 +325,8 @@ class AnalyticModel: threshold = self.attr_by_name[name][attr].stats.std_param_lut else: threshold = 0 - logger.debug(f"build_dtree({name}, {attr}, threshold={threshold})") - self.attr_by_name[name][attr].build_dtree( + logger.debug(f"build_rmt({name}, {attr}, threshold={threshold})") + self.attr_by_name[name][attr].build_rmt( threshold=threshold, ) else: @@ -369,9 +369,9 @@ class AnalyticModel: ): threshold = self.dtree_max_std[name][attr] logger.debug( - f"build_dtree({name}, {attr}, threshold={threshold})" + f"build_rmt({name}, {attr}, threshold={threshold})" ) - self.attr_by_name[name][attr].build_dtree( + self.attr_by_name[name][attr].build_rmt( threshold=threshold, ) else: diff --git a/lib/parameters.py b/lib/parameters.py index 0e09610..1ae4e4c 100644 --- a/lib/parameters.py +++ b/lib/parameters.py @@ -1004,7 +1004,7 @@ class ModelAttribute: ) return False - def build_dtree( + def build_rmt( self, with_function_leaves=None, with_nonbinary_nodes=None, @@ -1047,16 +1047,16 @@ class ModelAttribute: if loss_ignore_scalar and not with_function_leaves: logger.warning( - "build_dtree {self.name} {self.attr} called with loss_ignore_scalar=True, with_function_leaves=False. This does not make sense." + "build_rmt {self.name} {self.attr} called with loss_ignore_scalar=True, with_function_leaves=False. This does not make sense." ) relevance_threshold = float(os.getenv("DFATOOL_PARAM_RELEVANCE_THRESHOLD", 0.5)) logger.debug( - f"build_dtree(threshold={threshold}, relevance_threshold={relevance_threshold})" + f"build_rmt(threshold={threshold}, relevance_threshold={relevance_threshold})" ) - self.model_function = self._build_dtree( + self.model_function = self._build_rmt( self.param_values, self.data, with_function_leaves=with_function_leaves, @@ -1067,7 +1067,7 @@ class ModelAttribute: relevance_threshold=relevance_threshold, ) - def _build_dtree( + def _build_rmt( self, parameters, data, @@ -1291,7 +1291,7 @@ class ModelAttribute: child_parameters = list(map(lambda i: parameters[i], indexes)) child_data = list(map(lambda i: data[i], indexes)) assert len(child_data) - child[value] = self._build_dtree( + child[value] = self._build_rmt( child_parameters, child_data, with_function_leaves=with_function_leaves, |