lib/rumale/tree/gradient_tree_regressor.rb in rumale-0.12.5 vs lib/rumale/tree/gradient_tree_regressor.rb in rumale-0.12.6

@@ -212,15 +212,11 @@
         @leaf_weights.push(weight)
         @n_leaves += 1
         node
       end
 
-      def best_split(features, g, h, sum_g, sum_h)
-        order = features.sort_index
-        sorted_f = features[order].to_a
-        sorted_g = g[order].to_a
-        sorted_h = h[order].to_a
-        find_split_params(sorted_f, sorted_g, sorted_h, sum_g, sum_h, @params[:reg_lambda])
+      def best_split(f, g, h, sum_g, sum_h)
+        find_split_params(f.sort_index, f, g, h, sum_g, sum_h, @params[:reg_lambda])
       end
 
       def rand_ids
         @feature_ids.sample(@params[:max_features], random: @sub_rng)
       end