mnist数据集,运用Python实现softmax
该博客利用Python的mxnet库来简单实现softmax算法,源代码如下:主要包括训练函数,网络构成函数,损失函数,和预测函数。
import gluonbook as gb
from mxnet import autograd,nd
def softmax(x):
x_exp=x.exp()
partition=x_exp.sum(axis=1,keepdims=Tru...
原创
2018-12-07 15:26:09 ·
641 阅读 ·
0 评论