regress.py

著名的ldpc编解码的资料及源代码

"""
This module contains various regression tests designed to make sure
channel decoding functions are working correctly.
"""

import pycodes
from pycodes.utils import channels, CodeMaker, testing
from math import exp

(N,K) = (1200,600)
L = CodeMaker.make_H_gallager(N,3,6) 
E = (N-K)*6 # number of edges in the code graph

testing.RequireSuccessfulDecoding(
    'Small BSC test decoded with SumProductBP',
    N,K,E,L,'bit flips',0.05,channels.BSCOutput,numTrials=50,
    maxIter=20,algorithm='SumProductBP',requiredSuccessRate=0.8,
    verbose=0)

testing.RequireSuccessfulDecoding(
    'Small BSC test decoded with LRSumProductBP',
    N,K,E,L,'bit flips',0.05,
    lambda x,y: map(lambda z: exp(z),channels.BSCOutput(x,y)),
    numTrials=50,maxIter=20,
    algorithm='LRSumProductBP',requiredSuccessRate=0.8,verbose=0)