train_accuracy = np.mean(y_train == y_train_pred)源代码如上所示,我想问这个==是取对应相等的地方之后再做均值嘛?还是我根本就理解错了,这行代码到底是什么意思啊?为什么可以这么用?
2 回答
偶然的你
TA贡献1841条经验 获得超3个赞
mean是求平均值的意思,y_train == y_train_pred意思就是判断训练的值和预测的值是否相同,相等返回1,将相等的全部加起来/总训练数,就是训练集的准确率了,mean这里就是统计相等的做除法算出准确率的作用。所以 np.mean(y_train == y_train_pred)就是算训练集准确率的意思,如果改成np.mean(y_test == y_test_pred)就是算测试集的准确率,一般训练集和测试集准确率都要算,这样可以看出是否出现,训练集准确率很高,而测试集准确率却比较低,那么就会出现过拟合,我们需要两个都比较高
添加回答
举报
0/150
提交
取消