как получить данные о возможности от керас NN?

Question

Я изучаю keras NN с этим учебником

В этом примере мы классифицируем известный набор данных IRIS

model = Sequential()
model.add(Dense(16, input_shape=(4,)))
model.add(Activation('sigmoid'))
model.add(Dense(3))
model.add(Activation('softmax'))
model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=["accuracy"])
model.fit(train_X, train_y_ohe, epochs=100, batch_size=1, verbose=0);
loss, accuracy = model.evaluate(test_X, test_y_ohe, verbose=0)
print("Accuracy = {:.2f}".format(accuracy))

Однако я хотел бы распечатать матрицу возможностей для данных тестирования (т. Е. После применения активации («softmax»), какие шансы для данных тестирования принадлежат классам A, B, C, D). Может кто-нибудь сказать мне, как это сделать?

Answer 1

После обучения модели вы захотите применить ее к своему тестовому набору. Вы можете сделать:

class_probability = model.predict(test_X)

Проконсультируйтесь здесь :

предсказать (self, x, batch_size = None, verbose = 0, steps = None) Генерирует выходные прогнозы для входных выборок.