- fixed test_xor

- added test_linear
2026-01-03 13:07:08 +01:00
parent e56f99535a
commit e6c959d890
2 changed files with 49 additions and 3 deletions
@@ -2,9 +2,9 @@ import os.path

 from rbm.layer import Layer
 from rbm.status import Status
-from rbm.train import train, TrainingParams
+from rbm.train import train
 from rbm.matrix import Mat, np
-from rbm.entity import EntityParams
+from rbm.entity import EntityParams, TrainingParams

 WORK_DIR = "../../results"
 USE_OPTIMIZER = True
@@ -29,7 +29,7 @@ def xor():
 	training_batch = Mat([[0,1,1], [0,0,0], [1,1,0], [1,0,1]], dtype=np.float64)

 	# Train layer
-	train(layer.entity, training_batch, training_params, Status())
+	train(layer.entity, training_batch, Status())

 	# Save weights
 	layer.save(os.path.join(WORK_DIR, "xor_layer0_state.npz"))