tests/testthat/test-dbn-methods.R

In xrobin/DeepLearning: Deep Learning of neural networks

context("DBN methods")

l4g <- Layer(4, "gaussian")
l4b <- Layer(4, "binary")

l3g <- Layer(3, "gaussian")
l3c <- Layer(3, "continuous")


test_that("print works for DBN", {
	expect_output(print(DeepBeliefNet(l4g, l3g)))
	expect_output(print(DeepBeliefNet(l4g, l3g, l4b)))
	expect_output(print(DeepBeliefNet(l4g, l3g, l4b, l3c)))
})


test_that("Can print different stages of DBN", {
	expect_output(print(pretrained.mnist), "Pre-trained")
	expect_output(print(trained.mnist), "Fine-tuned")
})



test_that("drop works", {
	# Get a 1 layer DBN
	l1.dbn <- trained.mnist[1]
	expect_s3_class(l1.dbn, "DeepBeliefNet") # Not dropped yet
	expect_s3_class(drop(l1.dbn), "RestrictedBolzmannMachine") # Not dropped yet
})