如何求预测值为 1 且预测正确的样本个数?

2018-08-15 19:20:34 +08:00
 acone2003
有一个二分类问题,predict 为预测值,labels 为实际标签,请问一下如何显示预测值为 1 且预测正确的样本个数?
2240 次点击
所在节点    Python
5 条回复
acone2003
2018-08-15 21:35:43 +08:00
np.sum(predict == 1 and labels==1 )这样写不对,该怎么写呀?求助
nekota
2018-08-16 00:42:03 +08:00
result 是二维数组,[predict, label]
len(np.where(result==(1,1))[0])
diggerdu
2018-08-16 01:02:15 +08:00
np.sum(predict > 0.5 == label)
ivechan
2018-08-16 10:32:12 +08:00
没必要非得一条命令写吧?
在预测正确里,统计预测值为 1 的数量就行了。

或者你用 sklearn 之类的库的话,可以用库函数算出 confusion matrix。
这样 TP, TN, FP, FN 都有了。
acone2003
2018-08-16 16:51:17 +08:00
谢谢楼上三位,楼上的每个建议我都要查半天资料,长了不少知识----我是个初学者。
回 diggerdu: 用 np.sum(predict > 0.5 == label)这条语句怎么出现这个警告呀?
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/480156

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX