求pearson相关系数

aa = np.array([2,3,9,6,8])
bb = np.array([5,6,3,7,9])
train_pearson = np.corrcoef(aa, bb)[0][1]

train_pearson是一个2*2的矩阵，

train_pearson[0][1]即为aa和bb的Pearson score。


或者：

from scipy.stats import pearsonr

pearsonr(y_train, y_pred_train)[0]