diff --git a/src/mlpack/methods/decision_tree/decision_tree_impl.hpp b/src/mlpack/methods/decision_tree/decision_tree_impl.hpp index 306375d7c8..f5f832eddf 100644 --- a/src/mlpack/methods/decision_tree/decision_tree_impl.hpp +++ b/src/mlpack/methods/decision_tree/decision_tree_impl.hpp @@ -88,10 +88,11 @@ DecisionTree passlabels = tmpLabels; // Pass off work to the Train() method. arma::rowvec weights; // Fake weights, not used. - Train(tmpData, 0, tmpData.n_cols, tmpLabels, numClasses, weights, + Train(tmpData, 0, tmpData.n_cols, passlabels, numClasses, weights, minimumLeafSize, minimumGainSplit, maximumDepth, dimensionSelector); }