Style fix

This commit is contained in:
Roberto Hueso Gomez
2018-11-02 01:13:52 +01:00
parent 5594bb3f7b
commit 2b9d68a59e
@@ -181,21 +181,29 @@ static void mlpackMain()
arma::Row<double> weights =
std::move(CLI::GetParam<arma::Mat<double>>("weights"));
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
{
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?