TensorFlow/Keras binary_crossentropy损失函数


In [22]: y_true = [[0], [1]]                                                                                                                                                
In [23]: y_pred = [[0.9], [0.9]]                                                                                                                                            
In [24]: tf.keras.losses.binary_crossentropy(y_true, y_pred)                                                                                                                
Out[24]: <tf.Tensor: shape=(2,), dtype=float32, numpy=array([2.302584  , 0.10536041], dtype=float32)>
In [25]: y_true = [[0, 1], [1, 0]]                                                                                                                                          
In [26]: y_pred = [[0.9, 0.1], [0.9, 0.1]]                                                                                                                                  
In [27]: tf.keras.losses.binary_crossentropy(y_true, y_pred)                                                                                                                
Out[27]: <tf.Tensor: shape=(2,), dtype=float32, numpy=array([2.3025842 , 0.10536041], dtype=float32)>

所以不管是不是 one-hot encoding 都可以使用, 得到的 loss 是一样的.


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM