jens/pyRBM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

improved tests

Browse Source
This commit is contained in:
jens
2026-01-06 21:51:28 +01:00
parent 9025f54434
commit 074ded1581
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/tests/test_norbs.py
+2 -2
View File
@@ -16,7 +16,7 @@ class TestModel(Model):
else: else:
# Hidden binary # Hidden binary
self.unit1 = Entity((96 * 96, 333), EntityParams(do_gaussian_visible=True, do_gaussian_hidden=False), self.unit1 = Entity((96 * 96, 333), EntityParams(do_gaussian_visible=True, do_gaussian_hidden=False),
TrainingParams(learning_rate=0.001, momentum=0.9, num_epochs=1000, l2_lambda=0.01)) TrainingParams(learning_rate=0.001, momentum=0.9, num_epochs=1000, l2_lambda=0.001))
def forward(self, x: Mat): def forward(self, x: Mat):
x = self.unit1.forward(x) x = self.unit1.forward(x)
@@ -38,7 +38,7 @@ if __name__ == "__main__":
model.init(0.1) model.init(0.1)
# load state # load state
# model.load() model.load()
# Load train data # Load train data
train_batch = read_armadillo(os.path.join(prj_root, f"{prj_name}.training.dat")) train_batch = read_armadillo(os.path.join(prj_root, f"{prj_name}.training.dat"))