我想按照本教程中所述使用nsl.keras.GraphRegularization
训练二元目标深度神经网络模型。我的模型在中间密集层中具有三重半硬损失,不应"图正则化"。
从 Github 上的nsl.keras.GraphRegularization
定义:
将图正则化合并到
base_model
损失中。图正则化是在对数层上完成的,并且仅在训练期间完成。
这意味着中间三重半硬损失不会受到这种正则化的影响?
是的,没错。图正则化将仅应用于base_model
的输出。如果您的base_model
在另一层中使用三重半硬损耗,则该损失应不受影响并保留。如果不是这种情况,请在 https://github.com/tensorflow/neural-structured-learning/issues 提交错误。