configurable maximum stddev for kconfig model

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