From 77b36ddcbc919c8db25dc8d1366c2823348cd85e Mon Sep 17 00:00:00 2001
From: Vikas Shetty <shettyvikas209@gmail.com>
Date: Mon, 21 Oct 2019 10:19:44 +0530
Subject: [PATCH] Fixing test for callback

---
 src/mlpack/tests/callback_test.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/mlpack/tests/callback_test.cpp b/src/mlpack/tests/callback_test.cpp
index b0cd9d2d74..088028b9e1 100644
--- a/src/mlpack/tests/callback_test.cpp
+++ b/src/mlpack/tests/callback_test.cpp
@@ -126,10 +126,10 @@ BOOST_AUTO_TEST_CASE(LRWithOptimizerCallback)
     arma::mat data("1 2 3;"
                    "1 2 3");
     arma::Row<size_t> responses("1 1 0");
-    ens::StandardSGD sgd(0.005, 32, 500000, 1e-4);
+    ens::StandardSGD sgd(0.1, 1, 5);
     LogisticRegression<> logisticRegression(data, responses, sgd, 0.001);
     std::stringstream stream;
-    logisticRegression.Train<ens::L_BFGS>(
+    logisticRegression.Train<ens::StandardSGD>(
             data,
             responses,
             sgd,