vikramsoni2 · October 5, 2018 10:45
diff --git a/correlation_matrix.py b/correlation_matrix.py
 features = ['bedrooms', 'bathrooms', 'sqft_living',
            'sqft_lot', 'floors', 'waterfront', 'view', 'condition', 'grade',
            'sqft_above', 'sqft_basement', 'yr_built', 'yr_renovated', 'zipcode',
            'sqft_living15', 'sqft_lot15']

 f, ax = plt.subplots(figsize=(16, 12))
 plt.title('Pearson Correlation Matrix',fontsize=25)

 sns.heatmap(df[features].corr(), linewidths=0.25, vmax=1.0, square=True, cmap="BuGn_r", linecolor='k', annot=True)
	features = ['bedrooms', 'bathrooms', 'sqft_living',
	'sqft_lot', 'floors', 'waterfront', 'view', 'condition', 'grade',
	'sqft_above', 'sqft_basement', 'yr_built', 'yr_renovated', 'zipcode',
	'sqft_living15', 'sqft_lot15']

	f, ax = plt.subplots(figsize=(16, 12))
	plt.title('Pearson Correlation Matrix',fontsize=25)

	sns.heatmap(df[features].corr(), linewidths=0.25, vmax=1.0, square=True, cmap="BuGn_r", linecolor='k', annot=True)