动手深度学习3.4:softmax回归

softmax回归本身比较易懂,其公式是:![softmax](https://img-blog.csdnimg.cn/20200214195304159.png)
  • 主要作用我理解是通过取百分比,来做成了一个标准化的概率分布。下面对softmax函数做一个实现,本地用简单python实现了一遍,主要是做下课后题,代码如下:
import math

def softmax(a, b , c):
    sum = math.exp(a) +math.exp(b) +math.exp(c)
    i = math.exp(a) / sum
    j = math.exp(b) / sum
    k = math.exp(c) / sum
    return i, j, k

n = softmax(100, 101, 102)
print(n)
n1 = softmax(-2, -1, 0)
print(n1)

有个问题是我顺便验证abc=1000,1010,1020时报错,查网上的别人的博客改为.0后还是报错,看起来是数字溢出了。
然后用pytorch实现如下:

def softmax(X):
    X_exp = X.exp()
    partition = X_exp.sum(dim=1, keepdim=True)
    return X_exp / partition
X = torch.rand((2, 5))
X_prob = softmax(X)
print(X_prob, X_prob.sum(dim=1))

def net(X):
    return softmax(torch.mm(X.view((-1, num_inputs)), W)+ b)
y_hat = torch.tensor([[0.1, 0.3, 0.6], [0.3, 0.2, 0.5]])
y = torch.LongTensor([0, 2])
y_hat.gather(1, y.view(-1, 1))

def evaluate_accuracy(data_iter, net):
    acc_sum, n = 0.0, 0
    for X, y in data_iter:
        acc_sum += (net(X).argmax(dim=1) == y).float().sum().item()
        n += y.shape[0]
    return acc_sum / n
print(evaluate_accuracy(test_iter, net))
  1. 定义一个softmax函数,
    定义一个网络和准确度显示,
    最后使用d2l提供的train_ch3来训练模型,
    得到和原文差不多,需要注意的是复习了python的广播机制和交叉熵的推导,在知乎上找了一篇交叉熵和相对熵的文章,基本搞懂了。
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值