TensorFlow의 그래프 복원 - Tensor의 값 복원

Question

생명 보험 데이터의 가치를 예측하기 위해 인공 신경망을 구축했습니다. 그래프를 복원하면 예측 텐서를 가져 와서 내 가치를 볼 수 있습니다.

sess = tf.Session() 
new_saver = tf.train.import_meta_graph('model.ckpt.meta') 
new_saver.restore(sess, tf.train.latest_checkpoint('./')) 
graph = tf.get_default_graph() 
inputs = graph.get_tensor_by_name("inputs:0") 
predict_restore = graph.get_tensor_by_name("predicted:0") 
train_data = pd.read_csv(r"C:\...\tensorflow-1.3.1\tensorflow\train_titanic.csv") 
train_predict_restore = train_data.drop(["Survived"], axis=1) 
feed_dict={inputs:train_predict_restore} 
prob =sess.run(predict_restore,feed_dict) 

feed_dict에서 클라이언트의 속성을 텐서 입력에 넣었습니다. 이제 고객의 속성을 입력하는 함수를 작성하고 각각의 생존 확률 (prob)을 살펴 보겠습니다. tensorflow에서 하나 이상의 값을 검색하는 함수가 있습니까? (내 상황 텐서 입력의)는

Answer 1

나는가의 모양에있는 train_predict_restore [NUM_CUSTOMERS 속성] 생각합니다. 따라서 train_predict_restore [i]은 특정 고객이 ith을 나타냅니다. 당신이 뭔가를 할 수

, 여기

feed_dict={inputs:[train_predict_restore[i]]}//changed train_predict_restore to [train_predict_restore[i]] 
prob =sess.run(predict_restore,feed_dict) 

는, 출력은 번째 고객에 대한 확률 값입니다.

희망이 도움이됩니다.