精确度acc的计算

3层神经网络下手写数字识别

import sys, os
sys.path.append(os.pardir)  # 为了导入父目录的文件而进行的设定
import numpy as np
import pickle
from dataset.mnist import load_mnist
from common.functions import sigmoid, softmax

#获取训练数据,训练标签,测试数据,测试标签
#x_test.shape为(10000,784),y_test.shape为(10000,10)
def get_data():
    (x_train, t_train), (x_test, t_test) = load_mnist(normalize=True, flatten=True, one_hot_label=False)
    return x_test, t_test

#假设学习已经完成,所学到的参数被保存下来,,假设保存在同一目录下的sample_weight.pkl文件中,在预测阶段直接加载相应的模型即可。
def init_network():
    with open("sample_weight.pkl", 'rb') as f:
        network = pickle.load(f)
    return network

#预测
def predict(network, x):
    w1, w2, w3 = network['W1'], network['W2'], network['W3']
    b1, b2, b3 = network['b1'], network['b2'], network['b3']

    a1 = np.dot(x, w1) + b1
    z1 = sigmoid(a1)
    a2 = np.dot(z1, w2) + b2
    z2 = sigmoid(a2)
    a3 = np.dot(z2, w3) + b3
    y = softmax(a3)

    return y

#获得测试数据集
x, t = get_data()
#获取学习到的网络
network = init_network()

batch_size = 100 # 批数量
accuracy_cnt = 0

for i in range(0, len(x), batch_size):#一次处理100张图片,间隔100采样
    x_batch = x[i:i+batch_size] #X的形状是100x784
    y_batch = predict(network, x_batch) #y的形状是100 x 10
    p = np.argmax(y_batch, axis=1)#只对10这个维度里面的数据取最大值,并返回索引,返回p的维度为100
    accuracy_cnt += np.sum(p == t[i:i+batch_size])#比较神经网络预测解与正确标签,将回答正确的概率作为识别精度

print("Accuracy:" + str(float(accuracy_cnt) / len(x)))#len(x)为10000

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值