metro_shanghai.data.yml

# dataset and city
dataset: Metro
city: Shanghai

closeness_len: 6
period_len: 7
trend_len: 4
with_tpe: False

graph: Distance-Correlation-Line

data_range: all
train_data_length: all

threshold_distance: 5000
threshold_correlation: 0.7
threshold_interaction: 30

normalize: True
train: True

lr: 1e-5
early_stop_length: 200
early_stop_patience: 0.1
max_epoch: 20000
batch_size: 64
test_ratio: 0.1
MergeWay: sum

group: Shanghai
mark: BM