Minor changes to preprocess_split file

2020-10-10 19:37:16 +05:30
parent db94a2c14d
commit e6985fcda8
1 changed files with 1 additions and 1 deletions
@@ -148,7 +148,7 @@ static void mlpackMain()
        IO::GetParam<arma::Mat<size_t>>("input_labels");
    arma::Row<size_t> labelsRow = labels.row(0);

-    if(IO::HasParam("stratify"))
+    if(stratifyData)
    {
      const auto value =
          data::StratifiedSplit(data, labelsRow, testRatio, !shuffleData);