Pythonで東京の１日の新規感染者数を予想してみた！

こんにちは、ニコです。

今回は、Pythonを使って東京の１日の新規感染者数を予想してみました。

初めての機械学習ということで、シンプルなことしかできませんが、温かい目で見ていただけたら幸いです。

1. NumPyやPandasなどの外部ライブラリをインポート！

NumPy, Pandas, Matplotlibをそれぞれインポートします。

import matplotlib.dates as mdates

上記は、後にグラフを作る際に日にちを加えるためのコードです。

2. データを集める

都内の最新感染動向というホームページに、これまでの感染者数の推移が乗っているので、そこからデータを集めます。こちらがそのホームページです。

https://stopcovid19.metro.tokyo.lg.jp/

上記の画像は２月１４日から１２月２日までの１日の新規感染者数をまとめたデータです。NumPyを使って、ndarray化させています。

次にPandasのDataFrameを使って、綺麗な表を作ります。​

こんな感じですね。

3. 説明変数と目的変数を決め、訓練データとテストデータに分ける

ここではTechAcademyで習った技術を生かします。Xは、その日の感染者数、y は次の日の感染者数とします。ある日の感染者数がXなら、その次の日の感染者数はy になる」という形で X（説明変数）と y（目的変数）を設定します。

今回は、直近（後半の）３０日をテストデータにします。つまり、２月１４日から１１月３日までのデータを訓練データとし、１１月４日から１２月２日までをテストデータにします。

4. scikit-learnのLinearRegression（回帰）を使って、学習させる

今回は感染者数の予測を行いたいため、 linear_modelをインポートし、linear_model.LinearRegression()で学習をさせます。

predict()を使って、期待する性能が出たかを評価します

5. matplotlibを使って、グラフ化する

Pandasのpd.date_range()を使って、１１月４日から１２月２日までのデータを作ります。

あとはラベルやタイトルを入れて完成です。

結果はこうなりました！

赤い線はコンピューターが予想した数で、青い線は実際の数を表しています。

こうして見ると、ちゃんと予想はできてるように感じますが、専門家ではないので詳しい事はわからないです。

でも、初めての機械学習にしては上出来だったかなと思います。

もっと機械学習についての知識を深めて、難しいことにチャレンジしていけたらなと思います！

最後まで読んでいただきありがとうございました！