rbm_network_test.cpp File Reference

#include <mlpack/core.hpp>
#include <mlpack/methods/ann/init_rules/gaussian_init.hpp>
#include <mlpack/methods/ann/rbm/rbm.hpp>
#include <mlpack/methods/softmax_regression/softmax_regression.hpp>
#include <ensmallen.hpp>
#include "catch.hpp"

Include dependency graph for rbm_network_test.cpp:

Functions
	TEST_CASE ("BinaryRBMClassificationTest", "[RBMNetworkTest]")
	TEST_CASE ("ssRBMClassificationTest", "[RBMNetworkTest]")
template<typename MatType = arma::mat>
void	BuildVanillaNetwork (MatType &trainData, const size_t hiddenLayerSize)
	TEST_CASE ("MiscTest", "[RBMNetworkTest]")

Detailed Description

Author

Kris Singh

Shikhar Jaiswal

Tests the RBM Network

digits dataset source: {Lichman:2013 , author = "M. Lichman", year = "2013", title = "{UCI} Machine Learning Repository", url = "http://archive.ics.uci.edu/ml", institution = "University of California, Irvine, School of Information and Computer Sciences" }

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.