机器学习模型计算accuracy

濯君

于 2019-09-01 10:54:58 发布

阅读量4.5k

点赞数

CC 4.0 BY-SA版权

分类专栏：机器学习

本文链接：https://blog.csdn.net/zzldm/article/details/100181086

机器学习专栏收录该内容

28 篇文章

订阅专栏

本文深入探讨了机器学习中分类与回归问题的评估方法，详细介绍了准确率与均方误差的计算方式，以及如何利用这些指标来评估模型的预测性能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一：分类问题

from sklearn.metrics import accuracy_score
from sklearn.model_selection import train_test_split

X_train, X_test, y_train, y_test = train_test_split(train_x, train_y, test_size=0.33, random_state=123)
model.fit(X_train, y_train)
x_pred = xgbr.predict(X_test)
accuracy_score(y_test, y_pred)

二：回归问题

from sklearn.metrics import mean_squared_error
from sklearn.model_selection import train_test_split

X_train, X_test, y_train, y_test = train_test_split(train_x, train_y, test_size=0.33, random_state=123)
model.fit(X_train, y_train)
x_pred = xgbr.predict(X_test)
mean_squared_error(y_test, y_pred)

参考