Почему функция потерь не уменьшается при обучении Keras NN по собственной классификации? - PullRequest
0 голосов
/ 21 января 2019

Я пытаюсь отладить свой 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)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...