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

时间:2019-01-21 17:55:50

标签: python keras

我正在尝试调试NN。每当我使用其自己的分类但在使用其分类和另一个变量时,它就会收敛。它严重失败。我认为没有理由至少基于其分类不会收敛。

代码:

//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)

0 个答案:

没有答案