使用numpy计算准确率

准确率即预测正确个数占所有样本的比例,代码如下(数组中一行代表一个样本)

import numpy
preds = numpy.array([[1, 2, 3, 4],
                     [2, 2, 1, 1]])
true = numpy.array([[1, 2, 3, 4],
                    [2, 2, 2, 1]])
correct_prediction = numpy.equal(preds, true)
print(numpy.mean(correct_prediction))
>>> 0.875
发布了83 篇原创文章 · 获赞 4 · 访问量 5362

猜你喜欢

转载自blog.csdn.net/weixin_43486780/article/details/105389463