텐서 흐름 예측 결과를 .csv로 출력하는 방법은 무엇입니까?

Question

CNN을 사용하여 모델을 교육하고 있습니다.

여기 내 모델의 예측 부분입니다.

predictions = { 
    "classes": tf.argmax(input=logit2, axis=1), 
    "probabilities": tf.nn.softmax(logit2, name="softmax_tensor") 
} 

다음은 평가를 수행하는 주 코드입니다.

eval_input_fn = tf.estimator.inputs.numpy_input_fn(
    x={"x": images_test}, 
    y=test_labels, 
    num_epochs=1, 
    shuffle=False) 
eval_results = model.evaluate(input_fn=eval_input_fn) 

내가 (CSV 파일의 첫 번째 열에서) 테스트 이미지 이름의 목록을 가지고 있고, 내가 예측 및 출력을 두 번째 열에 해당 결과를 (만들고 싶어 지금 내 모델을 훈련했다 확률은 0에서 1 사이입니다.),이를 달성하는 방법 및 코드를 추가 할 위치는 무엇입니까?

미리 감사드립니다.

Answer 1

Estimator 클래스에는 예측을 반복 가능한 객체 (예 : scroll to the very bottom of this page)로 반환하는 predict function이 있습니다.

그래서 당신은 할 수 : csv로의 두 번째 컬럼에 쓰기에 관해서는

predictions = model.predict(input_fn=predict_input_fn) 
for p in predictions: 
    # write p['classes'] to the csv 

는 csv 파이썬 모듈을 살펴.