3dec3c177db7fb2c77eb38346a8d51075739d4d3,nt/speech_enhancement/merl_masks.py,,phase_sensitive_mask,#Any#Any#Any#,83

Before Change

    :param eps: Regularizing parameter to avoid division by zero.
    :return: Mask with shape (frames, 1, features)
    
    out_type = S.real.dtype
    Y = S + N
    theta = np.angle(S) - np.angle(Y)
    return (np.abs(S) / (np.abs(Y) + eps) * np.cos(theta)).astype(out_type)


def ideal_complex_mask(S, N):
     Calculates the ideal complex mask.

After Change

    :param eps: Regularizing parameter to avoid division by zero.
    :return: Mask with shape (frames, 1, features)
    
    return mask.phase_sensitive_mask(
        np.concatenate([S, N], 1),
        component_axis=1,
        eps=eps
    )[:, [0], :]


def ideal_complex_mask(S, N):
     Calculates the ideal complex mask.

Instances