diff options
Diffstat (limited to 'bin/analyze-archive.py')
| -rwxr-xr-x | bin/analyze-archive.py | 4 |
1 files changed, 3 insertions, 1 deletions
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) |