From e6985fcda8a76f19923175d2e3d990dcae98032b Mon Sep 17 00:00:00 2001
From: Anush V Kini <anushkini@gmail.com>
Date: Sat, 10 Oct 2020 19:37:16 +0530
Subject: [PATCH] Minor changes to preprocess_split file

---
 src/mlpack/methods/preprocess/preprocess_split_main.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/mlpack/methods/preprocess/preprocess_split_main.cpp b/src/mlpack/methods/preprocess/preprocess_split_main.cpp
index 2c01589e52..dfcdf8c640 100644
--- a/src/mlpack/methods/preprocess/preprocess_split_main.cpp
+++ b/src/mlpack/methods/preprocess/preprocess_split_main.cpp
@@ -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);