From b7c4b4e359a9c24fcce198fee7df2017b2a7d1a5 Mon Sep 17 00:00:00 2001 From: Andreas Tsouchlos Date: Mon, 7 Nov 2022 15:35:15 +0100 Subject: [PATCH] Changed default proximal decoder parameters --- sw/decoders/proximal.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/sw/decoders/proximal.py b/sw/decoders/proximal.py index 805ea9b..c702cff 100644 --- a/sw/decoders/proximal.py +++ b/sw/decoders/proximal.py @@ -9,8 +9,8 @@ class ProximalDecoder: # TODO: Is 'R' actually called 'decoding matrix'? # TODO: How large should eta be? # TODO: How large should step_size be? - def __init__(self, H: np.array, R: np.array, K: int = 100, step_size: float = 0.5, - gamma: float = 0.05, eta: float = 1.1): + def __init__(self, H: np.array, R: np.array, K: int = 100, step_size: float = 0.1, + gamma: float = 0.05, eta: float = 1.5): """Construct a new ProximalDecoder Object. :param H: Parity Check Matrix @@ -56,6 +56,7 @@ class ProximalDecoder: for k, x_k in enumerate(x): sum_result = 0 + for i in self._B[k]: prod = np.prod(x[self._A[i]]) sum_result += prod**2 - prod