update

Author: minmingzhu

mllib-dal/src/main/native/DecisionForestClassifierImpl.cpp

@@ -284,14 +284,25 @@ static jobject doRFClassifierOneAPICompute(
     t1 = std::chrono::high_resolution_clock::now();
     const auto result_train =
         preview::train(comm, df_desc, hFeaturetable, hLabeltable);
-    const auto result_infer =
-        preview::infer(comm, df_desc, result_train.get_model(), hFeaturetable);
-    jobject trees = nullptr;
     if (isRoot) {
         logger::println(logger::INFO, "Variable importance results:");
         printHomegenTable(result_train.get_var_importance());
         logger::println(logger::INFO, "OOB error:");
         printHomegenTable(result_train.get_oob_err());
+    }
+    const auto result_infer =
+        preview::infer(comm, df_desc, result_train.get_model(), hFeaturetable);
+
+    t2 = std::chrono::high_resolution_clock::now();
+    duration =
+        (float)std::chrono::duration_cast<std::chrono::milliseconds>(t2 -
+                                                                     t1)
+            .count();
+    logger::println(logger::INFO,
+                    "DF Classifier (native): training step took  %f secs.",
+                    duration / 1000);
+    jobject trees = nullptr;
+    if (isRoot) {
         logger::println(logger::INFO, "Prediction results:");
         printHomegenTable(result_infer.get_responses());
         logger::println(logger::INFO, "Probabilities results:\n");