From 016210b6c64881a683b45a91af213a1d1a1128ab Mon Sep 17 00:00:00 2001 From: Daniel Friesel Date: Thu, 9 Jul 2020 14:55:08 +0200 Subject: working candidate detection for decision trees --- bin/analyze-archive.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'bin/analyze-archive.py') diff --git a/bin/analyze-archive.py b/bin/analyze-archive.py index c531bb6..4dd2f90 100755 --- a/bin/analyze-archive.py +++ b/bin/analyze-archive.py @@ -112,7 +112,7 @@ import sys from dfatool import plotter from dfatool.loader import RawData, pta_trace_to_aggregate from dfatool.functions import gplearn_to_function -from dfatool.model import PTAModel +from dfatool.model import PTAModel, DecisionTreeModel from dfatool.validation import CrossValidator from dfatool.utils import filter_aggregate_by_param from dfatool.automata import PTA @@ -454,6 +454,8 @@ if __name__ == "__main__": pta=pta, ) + dtreemodel = DecisionTreeModel(by_name, parameters,) + if xv_method: xv = CrossValidator(PTAModel, by_name, parameters, arg_count) -- cgit v1.2.3