diff options
| author | Daniel Friesel <daniel.friesel@uos.de> | 2022-11-30 08:34:10 +0100 |
|---|---|---|
| committer | Daniel Friesel <daniel.friesel@uos.de> | 2022-11-30 08:34:10 +0100 |
| commit | 30d2faac5b27bb72e98d5f3ab57b20b1495c1ffa (patch) | |
| tree | 2d86f98593b6109ec5634b1ad372858873029c92 /lib | |
| parent | 1ec5b5340db1bc7959c38332481d0672aab1e029 (diff) | |
tree model complexity score: include number of non-leaf nodes
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/functions.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/lib/functions.py b/lib/functions.py index 032006e..6613e36 100644 --- a/lib/functions.py +++ b/lib/functions.py @@ -386,7 +386,7 @@ class SplitFunction(ModelFunction): def get_complexity_score(self): if not self.child: return 1 - ret = 0 + ret = 1 for v in self.child.values(): ret += v.get_complexity_score() return ret @@ -552,7 +552,7 @@ class CARTFunction(SKLearnRegressionFunction): return self.regressor.get_depth() def get_complexity_score(self): - return self.get_number_of_leaves() + return self.get_number_of_nodes() def to_json(self, feature_names=None, **kwargs): import sklearn.tree @@ -606,7 +606,7 @@ class LMTFunction(SKLearnRegressionFunction): return len(self.regressor._leaves.keys()) def get_complexity_score(self): - ret = 0 + ret = self.get_number_of_nodes() - self.get_number_of_leaves() for leaf in self.regressor._leaves.values(): ret += len( list( @@ -663,7 +663,7 @@ class XGBoostFunction(SKLearnRegressionFunction): return 1 + max(ret) def get_complexity_score(self): - return self.get_number_of_leaves() + return self.get_number_of_nodes() # first-order linear function (no feature interaction) |