def test_dbn_n_hidden_setter():
new_dbn = dbn.DBN()
try:
new_dbn.n_hidden = "a"
except:
new_dbn.n_hidden = (128, )
assert len(new_dbn.n_hidden) == 1
def test_dbn_models_setter():
new_dbn = dbn.DBN()
try:
new_dbn.models = 'a'
except:
new_dbn.models = []
assert len(new_dbn.models) == 0
def test_dbn_n_visible_setter():
new_dbn = dbn.DBN()
try:
new_dbn.n_visible = "a"
except:
new_dbn.n_visible = 1
assert new_dbn.n_visible == 1
try:
new_dbn.n_visible = 0
except:
new_dbn.n_visible = 1
assert new_dbn.n_visible == 1
def test_dbn_lr_setter():
new_dbn = dbn.DBN()
try:
new_dbn.lr = "a"
except:
new_dbn.lr = (0.1, )
assert len(new_dbn.lr) == 1
try:
new_dbn.lr = (0.1, 0.1)
except:
new_dbn.lr = (0.1, )
assert len(new_dbn.lr) == 1
def test_dbn_momentum_setter():
new_dbn = dbn.DBN()
try:
new_dbn.momentum = "a"
except:
new_dbn.momentum = (0, )
assert len(new_dbn.momentum) == 1
try:
new_dbn.momentum = (0, 0)
except:
new_dbn.momentum = (0, )
assert len(new_dbn.momentum) == 1
def test_dbn_T_setter():
new_dbn = dbn.DBN()
try:
new_dbn.T = "a"
except:
new_dbn.T = (0, )
assert len(new_dbn.T) == 1
try:
new_dbn.T = (0, 0)
except:
new_dbn.T = (0, )
assert len(new_dbn.T) == 1
def test_dbn_n_layers_setter():
new_dbn = dbn.DBN()
try:
new_dbn.n_layers = 0
except:
new_dbn.n_layers = 1
assert new_dbn.n_layers == 1
try:
new_dbn.n_layers = "a"
except:
new_dbn.n_layers = 1
assert new_dbn.n_layers == 1
def test_dbn_steps_setter():
new_dbn = dbn.DBN()
try:
new_dbn.steps = "a"
except:
new_dbn.steps = (1, )
assert len(new_dbn.steps) == 1
try:
new_dbn.steps = (1, 1)
except:
new_dbn.steps = (1, )
assert len(new_dbn.steps) == 1
def test_dbn_decay_setter():
new_dbn = dbn.DBN()
try:
new_dbn.decay = "a"
except:
new_dbn.decay = (0, )
assert len(new_dbn.decay) == 1
try:
new_dbn.decay = (0, 0)
except:
new_dbn.decay = (0, )
assert len(new_dbn.decay) == 1
def test_dbn_reconstruct():
test = torchvision.datasets.KMNIST(
root='./data',
train=False,
download=True,
transform=torchvision.transforms.ToTensor())
new_dbn = dbn.DBN(n_visible=784,
n_hidden=(128, 128),
steps=(1, 1),
learning_rate=(0.1, 0.1),
momentum=(0, 0),
decay=(0, 0),
temperature=(1, 1),
use_gpu=False)
e, v = new_dbn.reconstruct(test)
assert e >= 0
assert v.size(1) == 784
def test_dbn_steps():
new_dbn = dbn.DBN()
assert len(new_dbn.steps) == 1
def test_dbn_n_visible():
new_dbn = dbn.DBN()
assert new_dbn.n_visible == 128
def test_dbn_n_layers():
new_dbn = dbn.DBN()
assert new_dbn.n_layers == 1
def test_dbn_momentum():
new_dbn = dbn.DBN()
assert len(new_dbn.momentum) == 1
def test_dbn_n_hidden():
new_dbn = dbn.DBN()
assert len(new_dbn.n_hidden) == 1
def test_dbn_models_setter():
new_dbn = dbn.DBN()
new_dbn.models = []
assert len(new_dbn.models) == 0
def test_dbn_T():
new_dbn = dbn.DBN()
assert len(new_dbn.T) == 1
def test_dbn_decay():
new_dbn = dbn.DBN()
assert len(new_dbn.decay) == 1
def test_dbn_lr():
new_dbn = dbn.DBN()
assert len(new_dbn.lr) == 1
def test_dbn_models():
new_dbn = dbn.DBN()
assert len(new_dbn.models) == 1
