最近开始搭建神经网络,tensorflow是一个由google开发的很方便的深度学习框架,今天我将简要介绍如何使用softmax完成一个简单的分类网络搭建,也是一个tensorflow的入门操作。参看书籍《TensorFlow实战》
1.softmax
对于分类问题,一般采用softmax regression,即使是复杂的网络,如卷积或者循环,如果涉及到分类,在最后一层还是需要加softmax,表示为:
softmax(x)=normalize(exp(x))
判定为某一类的概率为:
softmax (x)i =exp( xi )/ ∑jexp ( xj
tensorflow之浅层(输入输出层)神经网络通过softmax分类mnist数据集
最新推荐文章于 2023-10-11 14:26:26 发布
本文介绍了使用tensorflow构建一个简单的softmax分类神经网络,以处理MNIST手写数字识别任务。通过加载Tensorflow库,导入MNIST数据集,设置 placeholders,定义weights和biases,实现Softmax Regression算法,设置损失函数和优化器,进行训练并验证模型的准确性。最后,展示了如何通过定义层添加函数来扩展网络。
摘要由CSDN通过智能技术生成