-
Notifications
You must be signed in to change notification settings - Fork 2
/
setting.py
51 lines (41 loc) · 1.29 KB
/
setting.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
"""
Setting for stock price model
"""
import enum
# https://www.infobyip.com/detectmonitordpi.php
DPI = 144
# SIZE OF INPUT IMAGE TO CNN
IMG_W = 112
IMG_H = 112
# THE DIRECTORY FOR STORING CHART DATA
CHART_DIR = "C:\\Users\\a18822\\Desktop\\StockPrice\\CHART"
# TO CACHE THE DATASET
CACHE_DIR = "C:\\Users\\a18822\\Desktop\\StockPrice\\CACHE"
class ChartType(enum.Enum):
BAR = "bar"
LINE = "line"
FLINE = "filled_line"
CANDLESTICK = "candlestick"
BAR_CANDLESTICK_FUSION = "bar_candlestick_fusion"
BAR_LINE_FUSION = "bar_line_fusion"
BAR_FLINE_FUSION = "bar_filled_line_fusion"
# tw_spydata_train.csv
TRAIN_SP500_DATA_FILE = "C:\\Users\\a18822\\Desktop\\StockPrice\\tw_spydata_train.csv"
TEST_SP500_DATA_FILE = "C:\\Users\\a18822\\Desktop\\StockPrice\\tw_spydata_test.csv"
VALIDATION_SP500_DATA_FILE = "C:\\Users\\a18822\\Desktop\\StockPrice\\tw_spydata_validation.csv"
ALPHA = 0.2
BETA = 1
GAMMA = 0.2
# When generating the dataset, please set it to False
# When training or testing: set it to True
ENABLE_IMAGE_DECODE = True
WINDOW_SIZE = 30
PREDICT_SIZE = 5
BATCH_SIZE = 32
EPOCH = 100
# NUMBER OF VALIDATION RECORD / BATCH_SIZE
VALIDATION_STEPS = int(9999 / BATCH_SIZE)
# VALIDATION_STEPS = int(500 / BATCH_SIZE)
NUMBER_OF_VALIDATION = 9999
LEARNING_RATE = 0.003
EPSILON = 0.1