深度学习 损失函数

分类问题和回归问题是监督学习的两大类。

分类问题常用的损失函数为交叉熵函数(cross entropy)

深度学习 损失函数

tensorflow实现过程:

# y为正确结果,y_pred为预测结果
loss = -tf.reduce_mean(y*tf.log(tf.clip_by_value(y_pred, 1e-10, 1.0)))

回归问题常用的损失函数为均方误差(MSE,mean squared error)

深度学习 损失函数

# y为正确结果,y_pred为预测结果
loss = tf.reduce_mean(tf.square(y-y))

 

上一篇:NLP(三十)利用ALBERT和机器学习来做文本分类


下一篇:几种模型评价指标实现代码