From b7f9bf66b0245d84705049fc3bb656b349ba28f9 Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Wed, 17 Dec 2025 20:05:36 +0100 Subject: [PATCH] use dot product for weights --- src/rbm/cd_train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/rbm/cd_train.py b/src/rbm/cd_train.py index 2f72377..58cb8c9 100644 --- a/src/rbm/cd_train.py +++ b/src/rbm/cd_train.py @@ -19,7 +19,7 @@ def cd_jens(v_states: np.ndarray, params: RbmParams, v_to_ph: Callable, h_to_pv: h_states = sample(h_probs) # Update weights (positive phase) - dw = np.vecmat(np.transpose(v_states), h_states) + dw = np.dot(np.transpose(v_states), h_states) dbv = np.sum(v_states, 0) dbh = np.sum(h_states, 0) @@ -37,7 +37,7 @@ def cd_jens(v_states: np.ndarray, params: RbmParams, v_to_ph: Callable, h_to_pv: h_probs = v_to_ph(v_probs) # Update weights (negative phase) - dw -= np.vecmat(np.transpose(v_probs), h_probs) + dw -= np.dot(np.transpose(v_probs), h_probs) dbv -= np.sum(v_probs, 0) dbh -= np.sum(h_probs, 0)