def get_corr_mask(corr, corr_thresh=0.8):
    """
    Mask the numbers below threshold in the corrlation matrix as 0.

    Parameters
    ----------
    corr: pandas dataframe of shape = [n_features, n_features]
    corr_thresh: float, default=0.8
        The correlation threshold above which a feature will be deemed correlated with
        another one and removed from the dataset.
    """
    corr_mask = corr.copy()
    corr_mask[corr_mask < corr_thresh] = 0
    return corr_mask