keras如何添加Lambda层以将张量的值四舍五入为0或1

Question

我想将Lambda层用作最后一层，以将S型激活的输出四舍五入为0或1。

但是似乎我不能只写这样的代码：

X = Lambda(lambda x: K.round(x))(X_output)

它将返回一个非派生错误。

如何修改它以将导数舍入为0或1？

Answer 1

您可以考虑对模型进行无round训练

model = Sequential()

model.add(Dense(32, input_shape=(5,)))
model.add(Dense(1, activation="sigmoid"))
#model.add(Lambda(lambda x: K.round(x)))

model.compile("adam", "mae") # or cross entropy
model.fit(X_train, y_train)

model.save_weights("/path/to/model.h5")

添加round了。

output = Lambda(lambda x: K.round(x)) (model.output)
m1 = Model(model.input, output)
m1.load_weights("/path/to/model.h5")