- initroduced types: BB-RBM GB-RBM, GG-RBM as property of entity

- training simplification and for clarity: added specialized cd_funcs for each entity type
2026-01-01 12:05:39 +01:00
parent 38b834c640
commit dc998d30a2
6 changed files with 119 additions and 8 deletions
@@ -40,7 +40,7 @@ if __name__ == "__main__":
 	test_batch = read_armadillo(os.path.join(prj_root, f"{prj_name}.test.dat"))

 	# Train
-	model.train(train_batch, TrainingParams(learning_rate=0.00001, momentum=0.9, do_rao_blackwell=True, num_epochs=100, num_gibbs_samples=3))
+	model.train(train_batch, TrainingParams(learning_rate=0.000001, momentum=0.9, do_rao_blackwell=False, num_epochs=1000, num_gibbs_samples=3))

 	# save state
 	model.save()