Как загрузить сохраненную модель оценки DNN для прогнозирования?

Question

У меня есть сохраненная модель оценщика DNN. Как я могу восстановить модель для прогнозирования? Я сохранил свою модель, используя:

#Create the input function
input_func = tf.estimator.inputs.pandas_input_fn(x=X_train, y=y_train, batch_size=100, num_epochs=None, shuffle=True)

#Create the model
model = tf.estimator.DNNClassifier(feature_columns=feat_cols, hidden_units=[10, 10], n_classes=2, model_dir=model_path)

Итак, внутри каталога model_path у меня есть:

• контрольная точка
• event.out.tf. ..
• graph.pbtxt
• model.ckpt-1 ...
• model.ckpt-1 ...
• model.ckpt-1 ...
• model.ckpt-5000 ...
• model.ckpt-5000 ...
• model.ckpt-5000 ...

Есть идеи?

Answer 1

Может быть, вы можете попробовать предиктор следующим образом.

from tensorflow.contrib import predictor

my_predict = predictor.from_saved_model(model_path)
prediction = my_predict({"x": [your_testing_data]})

Надеюсь, это поможет!