an.tsouchlos/ba-thesis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

Changed default proximal decoder parameters

Browse Source
This commit is contained in:
Andreas Tsouchlos 2022-11-07 15:35:15 +01:00
parent e6a7c39f50
commit b7c4b4e359
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
sw/decoders/proximal.py
View File

@ -9,8 +9,8 @@ class ProximalDecoder:
# TODO: Is 'R' actually called 'decoding matrix'? # TODO: Is 'R' actually called 'decoding matrix'?
# TODO: How large should eta be? # TODO: How large should eta be?
# TODO: How large should step_size 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, def __init__(self, H: np.array, R: np.array, K: int = 100, step_size: float = 0.1,
gamma: float = 0.05, eta: float = 1.1): gamma: float = 0.05, eta: float = 1.5):
"""Construct a new ProximalDecoder Object. """Construct a new ProximalDecoder Object.
:param H: Parity Check Matrix :param H: Parity Check Matrix
@ -56,6 +56,7 @@ class ProximalDecoder:
for k, x_k in enumerate(x): for k, x_k in enumerate(x):
sum_result = 0 sum_result = 0
for i in self._B[k]: for i in self._B[k]:
prod = np.prod(x[self._A[i]]) prod = np.prod(x[self._A[i]])
sum_result += prod**2 - prod sum_result += prod**2 - prod