riron1206 / candlestick_model

22 stars 8 forks source link

20日間のローソク足画像からCNNで株価の上下を予想するモデル

img_sample.png

ラベルの分け方かえていくつかモデル作ったが、以下のラベルを採用した

CM_without_normalize_optuna_best_trial_accuracy.png

行った手順

1. notebook/*.ipynb でデータ作成、モデル作成試す

2. 時系列の分け方でデータセット作成

train/test setに分割

$ python make_dataset.py -i D:\work\candlestick_model\output\orig_image_all_2day_label -o D:\work\candlestick_model\output\ts_dataset_all_2day_label --is_test_only --is_not_equalize

#### 3. code/tf_base_class*.py でモデル作成(パラメータチューニングも可能。少量データ版では試した)
```bash
$ python tf_base_class_all_data_2day_label.py -m train

4. bestモデルで予測

# JPX日経インデックス400 + 日経225 + 日経500種について
$ python tf_predict_best_model.py --is_2day_label

# 7269:スズキについて数日さかのぼって実行。下記は10日さかのぼる(2020/07/31からさかのぼって10日間毎日予測)
$ python tf_predict_best_model.py -c 7269 -d 2020-07-31 -t_d 10 --is_2day_label