Showing
2 changed files
with
7 additions
and
0 deletions
... | @@ -106,6 +106,9 @@ public class DecisionTree { | ... | @@ -106,6 +106,9 @@ public class DecisionTree { |
106 | (DecisionTreeRegressionModel) (model.stages()[1]); | 106 | (DecisionTreeRegressionModel) (model.stages()[1]); |
107 | System.out.println("Learned regression tree model:\n" + treeModel.toDebugString()); | 107 | System.out.println("Learned regression tree model:\n" + treeModel.toDebugString()); |
108 | 108 | ||
109 | + // save model | ||
110 | + model.save("./decisionTree.model"); | ||
111 | + | ||
109 | } | 112 | } |
110 | 113 | ||
111 | } | 114 | } | ... | ... |
src/main/java/detact/Main.java
0 → 100644
-
Please register or login to post a comment