前提・実現したいこと
機械学習を勉強している大学生です。
python,tensorflow,kerasを用いて2クラス分類を行うプログラムを書いています。
プログラムの大部分をkerasを用いて書いております。
分類確率をmodel.predictで出すことはできるのですが、
その確率値がどのデータに対するものなのかを出力する方法がわかりません。
以下は確率値を出すための部分的なプログラムです。
該当のソースコード
python
1result = model.predict(X_test) 2 3import csv 4with open('aaa.csv', 'w', newline='') as csv_file: 5 title = ['0', '1'] 6 writer = csv.writer(csv_file) 7 writer.writerow(title) 8 writer.writerows(result)
これを改良したいと思っているのですが、方法がわからずです。
理想としては
ファイル名 0である確率値 1である確率値
1.png 0.9999 0.0001
2.png 0.9999 0.0001
.
.
.
のように出力することです。
よろしくお願いします。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2018/11/13 08:04
2018/11/13 08:12
2018/11/13 08:50
2018/11/13 08:57
2018/11/13 09:20
2018/11/13 09:29 編集
2018/11/13 09:51 編集
2018/11/13 09:52
2018/11/13 10:15
2018/11/13 10:37