特征工程-相关性检验

# 绘制相关性矩阵热度图
fig,ax = plt.subplots(figsize=(18, 12))
sns.heatmap(corr_mat, xticklabels=True, yticklabels=True, square=False, linewidths=.5, annot=True, cmap='Oranges')
plt.savefig('相关性分析.jpg',dpi=400, bbox_inches='tight')
plt.show()

 

当通过热力图，查看到相关系数较高的变量时，还可以画图查看共线性具体状况，来判断是否需要删除一个。