如何解决为回归结果创建数据框表
我正在尝试创建一个函数来计算每个回归量的 R2 和 RMSE。
def apply_regressor_test(regressor,X_train,y_train,X_test,y_test,name):
r2= []
RMSE = []
reg_name = []
regressor.fit(X_train,y_train)
y_pred = regressor.predict(X_test)
cor = r2_score(y_test,y_pred)
r = np.sqrt(mean_squared_error(y_test,y_pred)
r2.append(cor)
RMSE.append(r)
reg_name.append(name)
df = pd.DataFrame({'R2': r2,'RMSE': RMSE},index = reg_name)
return df
我希望能够得到一个表格,将每个回归量的结果组合起来。我尝试创建另一个数据框来附加函数的结果。我最终得到的是这个。
regressor = LinearRegression()
df = apply_regressor_test(regressor,X_train_full_ss,y_train_full,X_test_ss,name = str(regressor))
df_results.append(df)
df_results
# This is what I get:
Out[306]:
[ R2 RMSE
LinearRegression() 0.643291 1.344363,R2 RMSE
RandomForestRegressor() 0.761544 1.099167,R2 RMSE
SVR() 0.706637 1.219165]
如何在只有一个列标题的表格中制作它?
谢谢
解决方法
作为@Raphael 在评论中的回答。我能够通过将返回的 dfs 连接起来来实现它,
pd.concat([df1,df2,df3],axis=0)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。