def feature_correlation(X, y, ax=None, method='pearson',
                        labels=None, sort=False, feature_index=None,
                        feature_names=None, **kwargs):
    viz = FeatureCorrelation(ax, method, labels, sort,
                             feature_index, feature_names, **kwargs)
    viz.fit(X, y, **kwargs)
    viz.finalize()
    return viz.ax