From 24c645137264c2f9d9146cda73677a67ec815ca3 Mon Sep 17 00:00:00 2001
From: Daniel Friesel <daniel.friesel@uos.de>
Date: Tue, 26 Oct 2021 15:14:37 +0200
Subject: allow custom standard deviation thresholds for decision tree
 compilation

---
 bin/analyze-kconfig.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

(limited to 'bin')

diff --git a/bin/analyze-kconfig.py b/bin/analyze-kconfig.py
index 004e691..533e621 100755
--- a/bin/analyze-kconfig.py
+++ b/bin/analyze-kconfig.py
@@ -40,6 +40,12 @@ def main():
         action="store_true",
         help="Build decision tree without checking for analytic functions first. Use this for large kconfig files.",
     )
+    parser.add_argument(
+        "--max-std",
+        type=str,
+        metavar="VALUE_OR_MAP",
+        help="Specify desired maximum standard deviation for decision tree generation, either as float (global) or <key>/<attribute>=<value>[,<key>/<attribute>=<value>,...]",
+    )
     parser.add_argument(
         "--export-model",
         type=str,
@@ -119,11 +125,29 @@ def main():
         # Release memory
         observations = None
 
+        if args.max_std:
+            max_std = dict()
+            if "=" in args.max_std:
+                for kkv in args.max_std.split(","):
+                    kk, v = kkv.split("=")
+                    key, attr = kk.split("/")
+                    if key not in max_std:
+                        max_std[key] = dict()
+                    max_std[key][attr] = float(v)
+            else:
+                for key in by_name.keys():
+                    max_std[key] = dict()
+                    for attr in by_name[key]["attributes"]:
+                        max_std[key][attr] = float(args.max_std)
+        else:
+            max_std = None
+
         model = AnalyticModel(
             by_name,
             parameter_names,
             compute_stats=not args.force_tree,
             force_tree=args.force_tree,
+            max_std=max_std,
         )
 
         if args.cross_validate:
@@ -135,6 +159,7 @@ def main():
                 parameter_names,
                 compute_stats=not args.force_tree,
                 force_tree=args.force_tree,
+                max_std=max_std,
             )
         else:
             xv_method = None
-- 
cgit v1.2.3