FlowPrediction/paras.py at master · onewalnut/FlowPrediction · GitHub

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
# -*- coding: utf-8 -*-
"""
-------------------------------------------------
   File Name:     paras
   Description :
   Author :       walnut
   date:          2021/1/5
-------------------------------------------------
   Change Activity:
                  2021/1/5:
-------------------------------------------------
"""
__author__ = 'walnut'


# Parameters of train and validate process
EPOCH = 500
TRAIN_BATCH = 30
TEST_BATCH = 10
LR = 0.5
LR_DECAY = 0.9
DECAY_PERIOD = 25
GPU_ID = "0"
GPU_IDS = [2, 3]

SAVE_FLAG = 10000000000000

# Parameters for sequence data and  RNN layers
INPUT_SIZE = 3
SEQ_LEN = 25
CALIBRATE_INTERVAL = 1


# temporalNet paras
LOCAL_HIDDEN_SIZE = 64
NUM_PART = 5
REGRESS_SIZE = 2048


# Data file
DATA_FILE = "D:/Workplace/PyCharm/FlowPrediction/Data/dataUse.json"
DATA_FILE_PATH = " "

RES_PATH = "D:/Workplace/PyCharm/FlowPrediction/Data/res/"