TensorFlow: How to predict from a SavedModel?

前端 未结 4 748
南方客
南方客 2020-12-25 15:41

I have exported a SavedModel and now I with to load it back in and make a prediction. It was trained with the following features and labels:

F1          


        
4条回答
  •  不知归路
    2020-12-25 15:57

    The constructor of tf.estimator.DNNClassifier has an argument called warm_start_from. You can give it the SavedModel folder name and it will recover your session.

提交回复
热议问题