我的模型总是给出相同的预测值

时间:2019-12-08 08:11:37

标签: machine-learning keras deep-learning

我使用gru层训练和预测了排名。 但是即使输入发生变化,它也始终提供相同的预测值。 例如, 输入(1,2,3,5)-> 6 输入(2,4,6,9)-> 6

我的模型出了什么问题?

model = Sequential()
model.add(GRU(32, batch_input_shape = (1,6,1), return_sequences=True, stateful=True))
model.add(GRU(16, return_sequences=True, stateful=True))
model.add(GRU(8, activation = 'relu', return_sequences = True, stateful = True))
model.add(GRU(4, activation ='relu', stateful = True))
model.add(Dense(1))
model.compile(optimizer='adam', loss='mae', metrics=['acc'])
model.fit(x[train], y[train], epochs=num_epochs, batch_size=1, verbose=0, callbacks=[early_stopping_callback, checkpointer],
           validation_data = (x[test], y[test]))

0 个答案:

没有答案