我正在尝试调试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)