신은섭(Shin Eun Seop)

save model

...@@ -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 }
......
1 +package detact;
2 +
3 +public class Main {
4 +}