手書きの数字認識問題Digit Recognizerの5回目(最終)の記事になります。
今回は、予測結果をKaggleに提出します。
前回記事のTorchvisionで用意されているResNet-18(畳み込みニューラル ネットワーク)モデルを使った成績が一番よかったので、このソースに提出ファイルの作成処理を追加します。
提出ファイルの作成
まずKaggle環境からtest.csvファイルをダウンロードしておきます。
前回のソースに下記のソースを追記し、実行します。(処理内容はコメントをご参照ください。)
[ソース]
1 | test_data = pd.read_csv('./test.csv', dtype=np.float32) |
submission.csvが出力されますので、このファイルをKaggleに提出します。
[提出結果]
提出結果は98.21%となりました。
十分な正解率になったかと思います。