Tensorflow入门2(模型训练,模型评估)

本文介绍了在TensorFlow中如何进行模型训练,利用交叉熵作为损失函数,并通过梯度下降法进行优化。同时,文章详细阐述了模型评估的过程,包括使用测试数据集预测并计算准确率。
摘要由CSDN通过智能技术生成

模型训练
接着上一节,模型已经用y = tf.nn.softmax(tf.matmul(x,W) + b)实现,现在来训练它,但是怎么才能知道训练得好不好呢?所以我们需要定义一个指标来评估模型的好坏,然后让这个指标最小,这个指标一般是成本(cost)或损失(loss),不过这两种方式其实也是一样的。常用的一个好用的成本函数叫交叉熵(cross-entropy)(感觉回到了高等热力学)。交叉熵产生于信息论里面的信息压缩编码技术,不过后来转到博弈论/机器学习等领域的评估手段,看下定义:
交叉熵定义
其中y是预测的概率分布,y’是实际的分布(就是输入的one-hot vector).可以把它理解为预测用于描述争相的效果如何的量度。这个概念还是很重要的。
为了计算交叉熵,做一个上面的y’,用y_代替

y_ = tf.placeholder("float", [None,10])

计算交叉熵:

cross_entropy = -tf.reduce_sum
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值