feedforward_network_2_test.cpp File Reference

#include <mlpack/core.hpp>
#include <mlpack/methods/ann/layer/layer.hpp>
#include <mlpack/methods/ann/loss_functions/mean_squared_error.hpp>
#include <mlpack/methods/ann/ffn.hpp>
#include <mlpack/methods/kmeans/kmeans.hpp>
#include <ensmallen.hpp>
#include "catch.hpp"
#include "serialization.hpp"
#include "custom_layer.hpp"

Include dependency graph for feedforward_network_2_test.cpp:

Functions
template<typename MatType = arma::mat, typename ModelType >
void	TestNetwork (ModelType &model, MatType &trainData, MatType &trainLabels, MatType &testData, MatType &testLabels, const size_t maxEpochs, const double classificationErrorThreshold)
	Train and evaluate a model with the specified structure.
	TEST_CASE ("RBFNetworkTest", "[FeedForwardNetworkTest]")
	Train the RBF network on a larger dataset.

Detailed Description

Author

Marcus Edel

Palash Ahuja

Tests the feed forward network.

mlpack is free software; you may redistribute it and/or modify it under the terms of the 3-clause BSD license. You should have received a copy of the 3-clause BSD license along with mlpack. If not, see http://www.opensource.org/licenses/BSD-3-Clause for more information.