From be64b0f8cee5efcc319d0cbbb738e8945db33984 Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Sat, 10 Jan 2026 16:46:27 +0100 Subject: [PATCH] refactored --- src/rbm/entity.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/rbm/entity.py b/src/rbm/entity.py index 1ab850c..b9c98ea 100644 --- a/src/rbm/entity.py +++ b/src/rbm/entity.py @@ -99,13 +99,13 @@ class Entity: # compute L1 term and penalize cost function (d_whv) l1_norm = np.sum(np.abs(self.state.w_hv)) - l1_term = params.l1_lambda*l1_norm*self.state.w_hv + l1_term = params.l1_lambda*l1_norm # compute L2 term and penalize cost function (d_whv) l2_norm = np.sum(np.square(self.state.w_hv)) - l2_term = params.l2_lambda*l2_norm*self.state.w_hv + l2_term = params.l2_lambda*l2_norm - self.grad.w_hv = params.momentum * self.grad.w_hv + lr * (d_whv-(l1_term+l2_term)) - lr * params.weight_decay * self.state.w_hv + self.grad.w_hv = params.momentum * self.grad.w_hv + lr * (d_whv - self.state.w_hv*(l1_term+l2_term)) - lr * params.weight_decay * self.state.w_hv return self.grad