diff options
Diffstat (limited to 'bin/analyze-archive.py')
| -rwxr-xr-x | bin/analyze-archive.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/bin/analyze-archive.py b/bin/analyze-archive.py index 65d80b3..5a6b8f0 100755 --- a/bin/analyze-archive.py +++ b/bin/analyze-archive.py @@ -115,7 +115,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 @@ -473,6 +473,9 @@ if __name__ == "__main__": pta=pta, ) + dtreemodel = DecisionTreeModel(by_name, parameters,) + print(dtreemodel.get_tree()) + if xv_method: xv = CrossValidator(PTAModel, by_name, parameters, arg_count) |