Pytorchで日経平均の予測2 - Qiita

どうもです。 つい先日後輩と銭湯に行く約束をしたものの、「すいません、女の子といて終電逃しました…」と断られ、怒りの余り作業が進み記事が完成しました。 ということで前回記事、Pytorchで日経平均の予測~幕間~で紹介した Using Deep Learning Neural Networks and Candlestick Chart Representation to Predict Stock Market についてのデータセットの作成、コーディング、学習を行ったので備忘録として残しておきます。 データセット作成 こちらで使われるデータセットはチャートのようなものですがちょっと特殊な画像です。（詳しくは論文をどうぞ！！） 論文を参考にして作成したものは このように４８×４８で上下が赤と緑によって表現されている画像になります。 以下にコードを紹介します。 # !/usr/bin/env