apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Avoid copies in decision tree main

Browse Source
This commit is contained in:
Roberto Hueso Gomez
2018-10-22 03:25:59 +02:00
parent 2530c45478
commit f4830fb3b5
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/decision_tree/decision_tree_main.cpp
+13 -4
View File
@@ -180,13 +180,22 @@ static void mlpackMain()
{
arma::Row<double> weights =
std::move(CLI::GetParam<arma::Mat<double>>("weights"));
model->tree = DecisionTree<>(trainingSet, model->info, labels,
numClasses, weights, minLeafSize, minimumGainSplit);
if (CLI::HasParam("print_training_error"))
model->tree = DecisionTree<>(trainingSet, model->info, labels,
numClasses, std::move(weights), minLeafSize, minimumGainSplit);
else
model->tree = DecisionTree<>(std::move(trainingSet), model->info,
std::move(labels), numClasses, std::move(weights), minLeafSize,
minimumGainSplit);
}
else
{
model->tree = DecisionTree<>(trainingSet, model->info, labels,
numClasses, minLeafSize, minimumGainSplit);
if (CLI::HasParam("print_training_error"))
model->tree = DecisionTree<>(trainingSet, model->info, labels,
numClasses, minLeafSize, minimumGainSplit);
else
model->tree = DecisionTree<>(std::move(trainingSet), model->info,
std::move(labels), numClasses, minLeafSize, minimumGainSplit);
}
// Do we need to print training error?