From 0f59ffb5f2ecb3dc23764cd566d962e483bf31e2 Mon Sep 17 00:00:00 2001 From: Daniel Friesel 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" -- cgit v1.2.3