相关疑难解决方法(0)

Keras只培训特定的产出

我正在使用带有张量流的Kears,我有一个3输出的模型,其中我只想训练2.

model = Model(input=input, output=[out1,out2,out3])
model.compile(loss=[loss1, loss2, loss3], optimizer=my_optimizer)

loss1(y_true, y_pred):
    return calculate_loss1(y_true, y_pred)

loss2(y_true, y_pred):
    return calculate_loss2(y_true, y_pred)

loss3(y_true, y_pred):
    return 0.0*K.mean(y_pred)
Run Code Online (Sandbox Code Playgroud)

我尝试用上面的代码来做,但我不确定它做了我想做的事情.因此,我认为它会增加损失,并且会根据损失对每个输出进行训练,同时我根本不想进行训练out3.(我需要out3因为它用于测试).谁能告诉我如何实现这一目标或让我放心,代码实际上是我想要的?

python neural-network keras

6
推荐指数
1
解决办法
2104
查看次数

标签 统计

keras ×1

neural-network ×1

python ×1