Changed default proximal decoder parameters

This commit is contained in:
Andreas Tsouchlos 2022-11-07 15:35:15 +01:00
parent e6a7c39f50
commit b7c4b4e359

View File

@ -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