получить список прогнозов нейронной сети

Question

Я создал нейронную сеть для классификации сообщений. Теперь я хочу собрать прогнозы в список в Python. Как мне это сделать?

Итак, вот модель:

model = Sequential()
model.add(layers.Dense(500, activation = "relu", input_shape=(7600,)))
# Hidden - Layers
model.add(layers.Dropout(0.4, noise_shape=None, seed=None))
model.add(layers.Dense(300, activation = "relu"))
model.add(layers.Dropout(0.4, noise_shape=None, seed=None))
model.add(layers.Dense(100, activation = "relu"))
model.add(layers.Dropout(0.4, noise_shape=None, seed=None))

model.add(layers.Dense(20, activation = "softmax"))
model.summary()

model.compile(loss="categorical_crossentropy",
              optimizer="adam",
              metrics=['accuracy'])



model.fit( np.array(vectorized_training), np.array(y_train_neralnet),
          batch_size=2000,
          epochs=3,
          verbose=1,
          validation_data=(np.array(vectorized_validation), np.array(y_validation_neralnet)))

Здесь я попытался напечатать форму validation_data, которая находится внутри метода model.fit (), но это даетошибка.

NameError: name 'validation_data' is not defined

Answer 1

Это то, что вы ищете:

preds = model.predict(X_test)