为生产服务Tensorflow模型

时间:2019-08-01 15:36:16

标签: python tensorflow machine-learning tensorflow-serving serving

我正在使用张量流估计器模型进行多标签文本分类。  做出预测后,我需要为模型提供预测服务,但是我没有获得正确的路径。

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)

我尝试了张量流服务,但是没有用。

0 个答案:

没有答案