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