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

Added forgotten import and fixed test

Browse Source
This commit is contained in:
Rishabh Garg
2021-07-12 10:04:24 +05:30
parent 80a094e545
commit 8d3a7d7799
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/tests/decision_tree_test.cpp
+3 -2
View File
@@ -13,6 +13,7 @@
#include <mlpack/methods/decision_tree/decision_tree.hpp>
#include <mlpack/methods/decision_tree/information_gain.hpp>
#include <mlpack/methods/decision_tree/gini_gain.hpp>
#include <mlpack/methods/decision_tree/mad_gain.hpp>
#include <mlpack/methods/decision_tree/random_dimension_select.hpp>
#include <mlpack/methods/decision_tree/multiple_random_dimension_select.hpp>
@@ -47,9 +48,9 @@ TEST_CASE("MADGainNormalTest", "[DecisionTreeRegressionTest")
arma::rowvec labels(10, arma::fill::randn); // Mean = 0.
// Theoretical gain.
const double theoreticalGain = 0.0;
double theoreticalGain = 0.0;
for (size_t i = 0; i < labels.n_elem; ++i)
theoreticalGain += std::abs(labels[i]);
theoreticalGain -= std::abs(labels[i]);
theoreticalGain /= (double) labels.n_elem;
// Calculated gain.