python - 为什么使用自己的分类训练 Keras NN 时损失函数不会减少？

Question

我正在尝试调试我的神经网络。每当我使用它自己的分类但是当我使用它的分类和它的另一个变量时，它就会收敛。它失败得很厉害。我认为没有理由至少根据它的分类它不会收敛。

代码：

//training variables
dummy_variable = ['Class', 'mBB']

model = Sequential()

model.add(Dense(4, activation='relu', input_dim=len(dummy_variable)))

model.add(Dense(1, activation='sigmoid'))

model.compile(optimizer='SGD', loss='binary_crossentropy', metrics=['accuracy'])

data = df_k1_2[dummy_variable].as_matrix()

labels = df_k1_2['Class'].as_matrix()

// Train the model, iterating on the data in batches of 32 samples
model.fit(data, labels, epochs=10, batch_size=32, shuffle=True)

标签： pythonkeras