forked from zepen/predict_Lottery_ticket
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.py
57 lines (49 loc) · 1.12 KB
/
config.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
52
53
54
55
56
57
# -*- coding: utf-8 -*-
"""
Author: BigCat
"""
import os
URL = "https://datachart.500.com/ssq/history/"
path = "newinc/history.php?start={}&end="
BOLL_NAME = [
("红球", "red"),
("蓝球", "blue")
]
train_data_path = "data/"
train_data_file = "data.csv"
# 模型相关参数
windows_size = 3
batch_size = 1
sequence_len = 6
red_n_class = 33
red_epochs = 1
red_embedding_size = 32
red_hidden_size = 32
red_layer_size = 1
blue_n_class = 16
blue_epochs = 1
blue_embedding_size = 32
blue_hidden_size = 32
blue_layer_size = 1
# 模型训练参数
red_learning_rate = 0.001
red_beta1 = 0.9
red_beta2 = 0.999
red_epsilon = 1e-08
blue_learning_rate = 0.001
blue_beta1 = 0.9
blue_beta2 = 0.999
blue_epsilon = 1e-08
# 模型路径
model_path = os.getcwd() + "/model/"
red_ball_model_path = model_path + "/red_ball_model/"
blue_ball_model_path = model_path + "/blue_ball_model/"
# 模型名
pred_key_name = "key_name.json"
red_ball_model_name = "red_ball_model"
blue_ball_model_name = "blue_ball_model"
extension = "ckpt"
# 日志路径
log_path = os.getcwd() + "/log/"
access_log = log_path + "access.log"
error_log = log_path + "error.log"