我正在使用张量流估计器模型进行多标签文本分类。 做出预测后,我需要为模型提供预测服务,但是我没有获得正确的路径。
eval_input_fn = tf.estimator.inputs.numpy_input_fn(
{'descriptions': np.array(test_descriptions).astype(np.str)},
test_encoded.astype(np.int32),
shuffle=False)
##"Code for saving the model"
import pickle
model.save('my_model.h5')
with open('tokenizer.pickle', 'wb') as handle:
pickle.dump(tokenizer, handle, protocol=pickle.HIGHEST_PROTOCOL)
##Saving Model part ends
estimator.evaluate(input_fn=eval_input_fn)
raw_text = ("Pemeriksaan pilihan makanan kita dan makanan yang kita masukkan "
"ke dalam tubuh kita. Berdasarkan memoar Jonathan Safran Foer.")
predict_input_fn = tf.estimator.inputs.numpy_input_fn(
{'descriptions': np.array([raw_text]).astype(np.str)},
shuffle=False)
prediction = estimator.predict(predict_input_fn)
for movie_genres in prediction:
top_2 = movie_genres['probabilities'].argsort()[-3:][::-1]
for genre in top_2:
text_genre = encoder.classes_[genre]
print(text_genre)
我尝试了张量流服务,但是没有用。