評価結果をCSVで出力する

Question

PythoとKerasで機械学習しました。
テストデータを使って、評価テストをします。
その結果を一枚一枚次のような形式でCSVファイルとして出力させたいです。
画像につけているファイル名、ラベル番号、評価したラベル、SOFTMAXの値
・・・・
なお、画像のデータは、ImagedataGeneratorで使うファイル構造でTESTに保存しています。

よろしくお願いします。

Comments

[t_obara]

一つ一つの結果を取得することはできているのですか？

[miyumiyumiyuki]

はい。できています。

[t_obara]

そのタイミングでprintするという話ではないのですか？

[miyumiyumiyuki]

そのタイミングではなく、testのファイルに格納している画像をバッチで読み込んで、その後にCSVへの掃き出しをしたいと思います。よろしくお願いします。

[t_obara]

結果は不要という意味？というか、「その後」にprintすれば良いのでは？　何が問題なのかがわかりません。もう少し噛み砕いてご教授いただければと思います。

[tiitoi]

「画像につけているファイル名、ラベル番号、評価したラベル、SOFTMAXの値」はとれていて、CSV に書き込む方法がわからないという質問でしょうか？

[miyumiyumiyuki]

はい。そうです。一方で、画像につけているファイル名がKERASでどのようにあつかわれているかわかりません。これも含めてCSVに吐き出したいです。

[tiitoi]

ImageDataGenerator はディレクトリからファイルを読み込んだ際のファイル名の情報は残していないので、ファイル名はとれないですよ。https://github.com/keras-team/keras-preprocessing/blob/master/keras_preprocessing/image.py の line 1920 行目参照。ファイル名が必要なのであれば、自分でディレクトリから読み込んで入力データを作るしかないです。

[miyumiyumiyuki]

わかりました。皆様、ご丁寧にありがとうございました。

Answer 1

Python標準のcsvライブラリを使用します

https://qiita.com/okadate/items/c36f4eb9506b358fb608

下記コードは参考程度に。。。実際動かしていません。上記URLを参考にしてください

import csv

test_data=[
["x","xx","xxx","xxxx","xxxxx"],
["y","yy","yyy","yyyy","yyyyy"]
]

with open('test.csv', 'w') as f:
    writer = csv.writer(f, lineterminator='\n') # 改行コードを指定
    writer.writerows(test_data) # 2次元配列の書き込み