Question

当我想使用tensorflow中的model.predict方法预测给定输入的用户结果时，我正在评论的imdb数据集中工作，它提供了一个数组。如何解释该数组的结果？

string = str(input())
new_string = token.texts_to_sequences(string)
padded_new_string = pad_sequences(new_string,maxlen = 120, truncating = 'post')
print(model.predict(padded_new_string))

我也使用了S型二进制分类

model = tf.keras.Sequential([
    tf.keras.layers.Embedding(10000,16,input_length = 120),
    tf.keras.layers.Flatten(),
    tf.keras.layers.Dense(12,activation = 'relu'),
    tf.keras.layers.Dense(6,activation = 'relu'),
    tf.keras.layers.Dense(1,activation = 'sigmoid')
    ])

任何帮助都会有所帮助

Answer 1

如果我没记错的话，model.predict会为numpy的每个相应预测返回y_test数组。要解释这些值，可以使用evaluate方法来查找模型的准确性

score = model.evaluate(x_test, y_test, verbose=0)
print('Test loss:', score[0])
print('Test accuracy:', score[1])

model.predict返回值

1 个答案: