我注意到在Keras中,优化器采用了表单的损失函数
def loss(y_true, y_pred):
//some loss function
如果我的损失函数只是模型参数的函数怎么办?如何在优化器中使用所述损失函数?以下是使用Keras优化器的示例:
sgd = optimizers.SGD(lr=0.001, decay=1e-6,
momentum=0.9, nesterov=True)
model.compile(optimizer=sgd,
loss=custom_objective,
metrics=['accuracy'])
def custom_objective(model.trainable_weights()):
//some loss function