将加权添加到keras中的输出

Question

我想在keras中为输出增加简单的权重。 我有这个模型：

model = keras.Sequential([
    keras.layers.Dense(32, input_shape=(1,))
    # Hidden layers
    keras.layers.Dense(2))
])

输入x是从1到0的随机浮点数 如果x> 0.9，则输出应为[0，1]，否则为[1，0]

但是如果我开始训练网络

x = np.random.random(1000)
y = (x > 0.9).astype(np.float32)
model.fit(x, y)

它只是学习输出：

   >>> model.predict(np.array([[1]]))<br>
   >>> [ [0.2] ]

所以我认为解决方案是对输出进行某种加权。我该怎么做？