diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/model.py | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/lib/model.py b/lib/model.py index 118a755..c74ebd1 100644 --- a/lib/model.py +++ b/lib/model.py @@ -1211,6 +1211,7 @@ class KConfigModel: def __init__(self, kconfig_benchmark, attribute): self.data = kconfig_benchmark.data self.symbols = kconfig_benchmark.symbols + self.max_stddev = 10 if callable(attribute): self.attribute = "custom" self.attr_function = lambda x: attribute(x[1]) @@ -1222,7 +1223,9 @@ class KConfigModel: self.attr_function = lambda x: x[1]["total"]["RAM"] else: raise ValueError("attribute must be a a function, 'rom', or 'ram'") - self.model = self.build_tree(self.symbols, self.data, 0) + + def build_tree(self): + self.model = self._build_tree(self.symbols, self.data, 0) def value_for_config(self, kconf): return self.model.model(kconf) @@ -1235,11 +1238,11 @@ class KConfigModel: } return output - def build_tree(self, this_symbols, this_data, level): + def _build_tree(self, this_symbols, this_data, level): rom_sizes = list(map(self.attr_function, this_data)) - if np.std(rom_sizes) < 100 or len(this_symbols) == 0: + if np.std(rom_sizes) < self.max_stddev or len(this_symbols) == 0: return self.Leaf(np.mean(rom_sizes), np.std(rom_sizes)) mean_stds = list() @@ -1280,8 +1283,8 @@ class KConfigModel: f"Level {level} split on {symbol} has {len(enabled)} children when enabled and {len(disabled)} children when disabled" ) if len(enabled): - node.set_child_y(self.build_tree(new_symbols, enabled, level + 1)) + node.set_child_y(self._build_tree(new_symbols, enabled, level + 1)) if len(disabled): - node.set_child_n(self.build_tree(new_symbols, disabled, level + 1)) + node.set_child_n(self._build_tree(new_symbols, disabled, level + 1)) return node |