Skip to content

Commit

Permalink
docs: README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
2018007956 committed Apr 21, 2024
1 parent 4af829e commit 2f19d5a
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions model_training/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
# Model Training
CNN 모델 학습을 위한 코드입니다.

## How to train CNN
1. yfinance를 사용한 stock_csv 생성
```
python data_generate/get_stock_data.py
```
2. (Optional) pickle data 생성
```
python data_generate/csv_to_pickle.py
```
3. config.json에서 csv_dir, pickle_dir 경로 및 하이퍼파라미터 지정
4. python train.py -c config.json

0 comments on commit 2f19d5a

Please sign in to comment.