junines的博客

原创 一种可以根据相关性进行特征筛选的函数

#去掉相关性太高的特征def identify_collinear(corr_matrix,correlation_threshold,X): #相关性矩阵 upper = corr_matrix.where(np.triu(np.ones(corr_matrix.shape), k = 1).astype(np.bool)) #相关性大小和阈值进行对比 to_drop = [column for column in upper.columns if any(upper[c...