3

15000からまでの値のリストがあります25000。(約) 20000 がカテゴリ 1 になり、残りがカテゴリ 2 になるように、それらを 2 つのカテゴリに分ける必要があります。そのために、ケラスで次のレイヤーを使用しています。

モデル = シーケンシャル()

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 になります。これを改善するにはどうすればよいですか?

4

2 に答える 2