Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
rfecv_3_correlations
correlated_features = set()
correlation_matrix = data.drop('Survived', axis=1).corr()
for i in range(len(correlation_matrix.columns)):
for j in range(i):
if abs(correlation_matrix.iloc[i, j]) > 0.8:
colname = correlation_matrix.columns[i]
correlated_features.add(colname)
@royinblr

This comment has been minimized.

Copy link

@royinblr royinblr commented May 17, 2020

It would be great if the column names inside correlated_features can be viewed.Is there any way to do so please?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment