建立了两个模型,结果rmse和r2完全一样
n_coefs = 4
model = [ols, ridge]
model_name = ['OLS', 'Ridge']
plt.figure(figsize = (10, 10))
for i in range(len(model)):
betas = model[i].coef_
indicies = np.argsort(np.abs(betas))[-n_coefs:]
top_predictors = np.array(predictors)[indicies]
top_betas = betas[indicies]
plt.subplot(2, 2, i+1)
plt.barh(top_predictors, top_betas, alpha=0.5, edgecolor='black')
sns.despine()
plt.xlabel('Beta coefficient')
plt.title(model_name[i]);
plt.suptitle('Figure . Top {} Beta Coefficients'.format(n_coefs), fontsize = 16)
plt.tight_layout()
最起码两个图应该不一样吧