From eb975da01884b7021ca23ca23d8de41560a8a09a Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Thu, 18 Dec 2025 07:56:08 +0100 Subject: [PATCH] refactored --- src/rbm/cd_train.py | 2 +- src/rbm/layer.py | 2 +- src/rbm/{helper.py => matrix.py} | 0 src/rbm/state.py | 6 +++--- 4 files changed, 5 insertions(+), 5 deletions(-) rename src/rbm/{helper.py => matrix.py} (100%) diff --git a/src/rbm/cd_train.py b/src/rbm/cd_train.py index 58cb8c9..eb27117 100644 --- a/src/rbm/cd_train.py +++ b/src/rbm/cd_train.py @@ -2,7 +2,7 @@ import numpy as np from collections.abc import Callable from params import RbmParams -from helper import sample, gaussian +from matrix import sample, gaussian def cd_jens(v_states: np.ndarray, params: RbmParams, v_to_ph: Callable, h_to_pv: Callable): v_probs = v_states diff --git a/src/rbm/layer.py b/src/rbm/layer.py index 55a19bd..4b811d2 100644 --- a/src/rbm/layer.py +++ b/src/rbm/layer.py @@ -2,7 +2,7 @@ import numpy as np from collections.abc import Callable from params import RbmParams from state import RbmState -from helper import sample, prob, uniform, rms_error_accu +from matrix import sample, prob, rms_error_accu from status import Status from cd_train import cd_jens diff --git a/src/rbm/helper.py b/src/rbm/matrix.py similarity index 100% rename from src/rbm/helper.py rename to src/rbm/matrix.py diff --git a/src/rbm/state.py b/src/rbm/state.py index 9e757cb..d53e89c 100644 --- a/src/rbm/state.py +++ b/src/rbm/state.py @@ -1,5 +1,5 @@ import numpy as np -from helper import uniform +from matrix import uniform class RbmState: @@ -33,10 +33,10 @@ class RbmState: return obj def v_to_h(self, visible: np.ndarray) -> np.ndarray: - return np.vecmat(visible, self.w_hv) + self.b_h + return np.dot(visible, self.w_hv) + self.b_h def h_to_v(self, hidden: np.ndarray) -> np.ndarray: - return np.vecmat(hidden, np.transpose(self.w_hv)) + self.b_v + return np.dot(hidden, np.transpose(self.w_hv)) + self.b_v def to_file(self, filename: str): np.savez(filename, whv=self.w_hv, bv=self.b_v, bh=self.b_h)