Как я могу получить прогноз от h5 с моделью тензорного потока

Question

Таким образом, я не могу понять, как получить показатель пригодности в% или метку из этой матрицы.

История создания этой модели:

- Модель должнаопределить, следует ли удалить комментарий (/ text) или нет.

- метки имеют значение True или False

Есть ли способ заставить его работать или как я могу преобразовать этот вектор?

Мой код:

model = models.load_model(h5_model)
string = "sh!t"
x_test = string.lower()
tok = text.Tokenizer(num_words=max_features, lower=True)
tok.fit_on_texts(list(x_test))
x_test = tok.texts_to_sequences(x_test)
text_preprocessing_for_single_comments(string)
prediction = model.predict(to_predict)
print(prediction)

>>[[0.5180945 ][0.5354299 ][0.47555092] [0.5636673]]

Answer 1

прогнозы являются вероятностями классов.Для преобразования в процентах:

prediction *= 100

Если 1 означает Истина, а 0 означает Ложь, то вероятность выше 0,5 будет означать Истина, а ниже - ложь.

prediction = np.round( prediction )
# Get an array of 0s and 1s
class = np.argmax( prediction )