working candidate detection for decision trees

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)