<p>Btw. I used the following code to test:</p>

<pre><code>auto GetVanillaNetwork(arma::mat&amp; trainData, arma::mat&amp; trainLabels)
{
  int hiddenLayerSize = 10;

  LinearLayer&lt;&gt; inputLayer(trainData.n_rows, hiddenLayerSize);
  BiasLayer&lt;&gt; inputBiasLayer(hiddenLayerSize);
  BaseLayer&lt;&gt; inputBaseLayer;

  LinearLayer&lt;&gt; hiddenLayer1(hiddenLayerSize, trainLabels.n_rows);
  BiasLayer&lt;&gt; hiddenBiasLayer1(trainLabels.n_rows);
  BaseLayer&lt;&gt; outputLayer;

  BinaryClassificationLayer classOutputLayer;

  auto modules = std::make_tuple(inputLayer, inputBiasLayer, inputBaseLayer,
                          hiddenLayer1, hiddenBiasLayer1, outputLayer);

  FFN&lt;decltype(modules), decltype(classOutputLayer), RandomInitialization,
      MeanSquaredErrorFunction&gt; net(modules, classOutputLayer);

  RMSprop&lt;decltype(net)&gt; opt(net, 0.01, 0.88, 1e-8,
      20 * trainData.n_cols, 1e-18);

  return net;
}

BOOST_AUTO_TEST_CASE(VanillaNetworkTest)
{
  // Load the dataset.
  arma::mat dataset;
  data::Load("thyroid_train.csv", dataset, true);

  arma::mat trainData = dataset.submat(0, 0, dataset.n_rows - 4,
      dataset.n_cols - 1);
  arma::mat trainLabels = dataset.submat(dataset.n_rows - 3, 0,
      dataset.n_rows - 1, dataset.n_cols - 1);

  data::Load("thyroid_test.csv", dataset, true);

  arma::mat testData = dataset.submat(0, 0, dataset.n_rows - 4,
      dataset.n_cols - 1);
  arma::mat testLabels = dataset.submat(dataset.n_rows - 3, 0,
      dataset.n_rows - 1, dataset.n_cols - 1);


  auto net = GetVanillaNetwork(trainData, trainLabels);
  arma::mat prediction;
  net.Predict(dataset, prediction);
}
</code></pre>

<p style="font-size:small;-webkit-text-size-adjust:none;color:#666;">&mdash;<br>Reply to this email directly or <a href="https://github.com/mlpack/mlpack/issues/531#issuecomment-191289070">view it on GitHub</a>.<img alt="" height="1" src="https://github.com/notifications/beacon/AJ4bFKYehn72Oa395hShSPShei1M30iFks5ppa3IgaJpZM4HmbCC.gif" width="1" /></p>
<div itemscope itemtype="http://schema.org/EmailMessage">
<div itemprop="action" itemscope itemtype="http://schema.org/ViewAction">
  <link itemprop="url" href="https://github.com/mlpack/mlpack/issues/531#issuecomment-191289070"></link>
  <meta itemprop="name" content="View Issue"></meta>
</div>
<meta itemprop="description" content="View this Issue on GitHub"></meta>
</div>