diff --git a/src/rbm/train.py b/src/rbm/train.py index ea183e8..0573dab 100644 --- a/src/rbm/train.py +++ b/src/rbm/train.py @@ -94,7 +94,7 @@ def cd_binary_binary(entity: Entity, data_pos: Mat): def cd_gaussian_binary(entity: Entity, data_pos: Mat): # Positive phase - h_probs_pos = prob(entity.h_given_v(data_pos + sample_gaussian(data_pos))) + h_probs_pos = prob(entity.h_given_v(data_pos)) # Update weights (positive phase) dw = np.dot(np.transpose(data_pos), h_probs_pos)