attitudechunfeng

def gen_probs(n, a, b):
    probs = np.zeros(n+1)
    for k in range(n+1):
        probs[k] = binom(n, k) * beta(k + a, n - k + b) / beta(a, b)
    return probs

class DCA(nn.Module):
    def __init__(self, attn_dim):
        super().__init__()
        self.conv_static = nn.Conv1d(1, 8, padding=(21 - 1) // 2, kernel_size=21, bias=True)