From 0f59ffb5f2ecb3dc23764cd566d962e483bf31e2 Mon Sep 17 00:00:00 2001
From: Daniel Friesel <daniel.friesel@uos.de>
Date: Wed, 13 Oct 2021 16:06:30 +0200
Subject: analyze-kconfig: add cross-validation support

---
 lib/model.py | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'lib')

diff --git a/lib/model.py b/lib/model.py
index 590c6f0..06ff25c 100644
--- a/lib/model.py
+++ b/lib/model.py
@@ -75,6 +75,7 @@ class AnalyticModel:
         function_override=dict(),
         use_corrcoef=False,
         compute_stats=True,
+        force_tree=False,
     ):
         """
         Create a new AnalyticModel and compute parameter statistics.
@@ -134,6 +135,13 @@ class AnalyticModel:
         if compute_stats:
             self._compute_stats(by_name)
 
+        if force_tree:
+            for name in self.names:
+                for attr in self.by_name[name]["attributes"]:
+                    # TODO specify correct threshold
+                    self.build_dtree(name, attr, 0)
+            self.fit_done = True
+
     def __repr__(self):
         names = ", ".join(self.by_name.keys())
         return f"AnalyticModel<names=[{names}]>"
-- 
cgit v1.2.3