code: 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