From f29baecfbfd487cbccae7fb16676dd4a73485541 Mon Sep 17 00:00:00 2001
From: Anmolpreet Singh <54476451+Anmol2001@users.noreply.github.com>
Date: Sat, 2 Jan 2021 12:39:04 +0530
Subject: [PATCH] Update feedforward_network_test.cpp

I have changed the test network with stochastic option to be off so that randomization was leading to different predictions.
---
 src/mlpack/tests/feedforward_network_test.cpp | 27 +++----------------
 1 file changed, 4 insertions(+), 23 deletions(-)

diff --git a/src/mlpack/tests/feedforward_network_test.cpp b/src/mlpack/tests/feedforward_network_test.cpp
index 6d53b76b20..acb99a063f 100644
--- a/src/mlpack/tests/feedforward_network_test.cpp
+++ b/src/mlpack/tests/feedforward_network_test.cpp
@@ -167,36 +167,17 @@ TEST_CASE("CheckCopyMovingReparametrizationNetworkTest", "[FeedForwardNetworkTes
 
   /*
    * Construct a feed forward network with trainData.n_rows input nodes,
-   * hiddenLayerSize hidden nodes and trainLabels.n_rows output nodes. The
-   * network structure looks like:
-   *
-   *  Input         Hidden        Output
-   *  Layer         Layer         Layer
-   * +-----+       +-----+       +-----+
-   * |     |       |     |       |     |
-   * |     +------>|     +------>|     |
-   * |     |     +>|     |     +>|     |
-   * +-----+     | +--+--+     | +-----+
-   *             |             |
-   *  Bias       |  Bias       |
-   *  Layer      |  Layer      |
-   * +-----+     | +-----+     |
-   * |     |     | |     |     |
-   * |     +-----+ |     +-----+
-   * |     |       |     |
-   * +-----+       +-----+
+   * followed by a linear layer and then a reparametrization layer
    */
 
   FFN<NegativeLogLikelihood<> > *model = new FFN<NegativeLogLikelihood<> >;
-  model->Add<Linear<> >(trainData.n_rows, 8);
-  model->Add<SigmoidLayer<> >();
-  model->Add<Reparametrization<> >();
+  model1->Add<Linear<> >(trainData.n_rows, 8);
+  model1->Add<Reparametrization<> >(4,false,true,1);
   model->Add<LogSoftMax<> >();
 
   FFN<NegativeLogLikelihood<> > *model1 = new FFN<NegativeLogLikelihood<> >;
   model1->Add<Linear<> >(trainData.n_rows, 8);
-  model1->Add<SigmoidLayer<> >();
-  model1->Add<Reparametrization<> >();
+  model1->Add<Reparametrization<> >(4,false,true,1);
   model1->Add<LogSoftMax<> >();
 
   // Check whether copy constructor is working or not.