diff --git a/spark_/train_spark_RF.java b/spark_/train_spark_RF.java index 84d741f..a04404e 100644 --- a/spark_/train_spark_RF.java +++ b/spark_/train_spark_RF.java @@ -98,6 +98,7 @@ public static void main(String[] args) { System.out.println("----------------- accuracy ----------------- "); System.out.println("Trained RF model:\n" + rfModel.toDebugString()); + System.out.println("accuracy: " + rmse ); System.out.println("----------------- accuracy ----------------- "); } } \ No newline at end of file