Inception v3网络的标签平滑

时间:2019-01-26 08:26:09

标签: python tensorflow keras

inception v3论文中,他们讨论了关于label smoothing的问题,但是在(keras代码)here中,没有提及标签平滑。如何为Inception v3网络编写用于标签平滑的代码。建议将其包括在损失函数中。我正在使用categorical crossentropy损失函数。因此,如何使自定义categorical crossentropy用于标签平滑。

1 个答案:

答案 0 :(得分:1)

您可以将label_smoothing参数传递给函数CategoricalCrossentropy以进行标签平滑。

import tensorflow as tf

# when compiling the model
model.compile('sgd', loss=tf.keras.losses.CategoricalCrossentropy(label_smoothing=0.1))