- 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
@@ -26,7 +26,7 @@ class Model(ABC):
 			params = [params]*len(entities)
 		_batch = np.copy(batch)
 		for entity, param in zip(entities, params):
-			train(entity, _batch, param, Status(), cd_jens)
+			train(entity, _batch, param, Status())
 			_batch = entity.forward(_batch, num_gibbs=param.num_gibbs_samples)

 	@abstractmethod