(data, col=[])
return img
correlation matrix for plot roc

    :param data: is the data to find correlations for
    :param col: is the specific column you want to find correlations to
    :return correlation matrix
correlation matrix for plot roc
correlation matrix for plot roc
def plot_corr(data, col=[]):
    '''
    correlation matrix for plot roc

    :param data: is the data to find correlations for
    :param col: is the specific column you want to find correlations to
    :return correlation matrix
    '''

    img = plt.figure()
    if col:
        data = data.loc[:, data.columns.intersection(['a', 'b'])]
    corr = data.corr()
    sns.heatmap(
        corr,
        vmin=-1, vmax=1, center=0,
        square=True)
    return img
def plot_corr ( data , col = [ ] ) : img = plt . figure ( ) if col : data = data . loc [ : , data . columns . intersection ( [ 'a' , 'b' ] ) ] corr = data . corr ( ) sns . heatmap ( corr , vmin = - 1 , vmax = 1 , center = 0 , square = True ) return img