我有一个值列表,范围从15000
到25000
。我必须将它们分为两类,这样(大约)20000 将最终归入类别 1,其余的归入类别 2。我发现 sigmoid 激活应该适用于此。为此,我在 keras 中使用了以下层:
模型=顺序()
model.add(Dense(1 , input_dim =1 ))
model.add(Activation('sigmoid'))
model.add(Dense(2 , init='normal' , activation = 'softmax'))
model.compile(loss='mean_absolute_error', optimizer='rmsprop')
model.fit(X_train, y_train, validation_data=(X_test, y_test),epochs=10,batch_size=200,verbose=2)
但是,当我为我的示例案例运行模型时,所有值都归入类别 2。我该如何改进呢?