［初心者です］MNISTのモデルをCSVファイルに出力する方法教えてください
Kerasを使ってMNISTのモデルを作りました。CSVファイルで提出する必要があるのですが方法がわかりません。ご教示頂けますでしょうか。
0 likes
3Answer
CSVファイルに記載する項目は何でしょうか？例えば、テストデータの真値（正解ラベル）と分類結果、信頼度を保存するなら、以下のようにします。
import pandas as pd
pred = model.predict(X_test) # shape == (10000, 10)
y_pred = np.argmax(pred, axis=1) # shape == (10000,)
confidence = np.max(pred, axis=1) # shape == (10000,)
df = pd.DataFrame({
'y_test': np.argmax(y_test, axis=1),
'y_pred': y_pred,
'confidence': confidence,
})
df.to_csv("prediction.csv", index=False)
出力例は以下のとおりです（5行目以降は省略）。
prediction.csv
y_test,y_pred,confidence
7,7,0.9996538
2,2,0.9999999
1,1,0.9993451
1Like
Comments
@kimukazu1029
Questioner
コメントありがとうございます。自分の学が足りず，重ね重ね質問をしてしまう形となってしまい申し訳ないのですが，書いていただいたコードを実行してもcsvファイルが保存された形跡がありませんでした。自分のやり方が間違っているのでしょうか...- 質問本文のコードをすべて実行した後に、私がコメントで書いたコードを実行してください（hist = model.fit(... の後ろに、私が書いたコードを追加して実行）。
0Like
Google Colaboratoryでの実行結果のスクリーンショットです。左端の「ファイル」から、現在のディレクトリに prediction.csv が保存されていることが確認できます。あるいは、ファイルの保存先を、自分で把握できているディレクトリに変更してみてください。
0Like
Comments
@kimukazu1029
Questioner
ありました，，，ありがとうございます！