tensorflow - keras 将输出层的值限制为值向量

Question

我的数组X输入为：

[[12,13,14],
..........
[32,15,19],
[23,16,17]
]

我的输出y为：

[[5]
..........
[7],
[9]
]

我确信 y 在所有情况下都只取值 5,7,9

谁能告诉我哪种型号最适合我使用。目前我使用过：

  model = Sequential()
  model.add(Dense(len(sequence)*10, input_shape=(len(sequence),), activation='relu')) 
  model.add(Dense(1, activation='relu')) 
  model.compile(loss='mse', optimizer='adam')

然后当我适合：

yhat1 = md1.predict(xCheck)

我得到的输出为 3、5、11，这不适合我。

标签： tensorflowkeraskeras-layer

Answer 1

是的，我明白了。如果您按照我告诉您的方式更改最后一层和标签，您将获得所需的输出。否则，按照您的方法，唯一的方法就是训练更多。我建议你多看看这个理论。