我有一個值列表,範圍從15000
到25000
。我必須將它們分成兩類,(大約)20000最終將歸入第1類,其餘的歸類爲第2類。我發現S形激活應該適用於此。我使用了以下幾層keras爲: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類結束了我怎樣才能改進這個?
你訓練了你的模型嗎? –
這將有助於您發佈整個代碼。 –
需要查看'model.complie()','model.fit()'和X和Y值的數據樣本 – DJK