From 0febd4e6b6afc8b87bd86f3a91a61cc5f45abe87 Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Sun, 7 Jun 2026 17:48:22 +0200 Subject: [PATCH] Rbm::Train(cd_bb): added do_gibbs_sample_hidden --- src/rbm/train.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/rbm/train.py b/src/rbm/train.py index 25cfc2d..fafed93 100644 --- a/src/rbm/train.py +++ b/src/rbm/train.py @@ -49,6 +49,9 @@ def cd_binary_binary(entity: Entity, data_pos: Mat): # Gibbs sampling for _ in range(params.num_gibbs_samples-1): + if params.do_gibbs_sample_hidden: + h_probs_neg = sample(h_probs_neg) + data_neg = prob(entity.v_given_h(h_probs_neg)) h_probs_neg = prob(entity.h_given_v(data_neg))