nnzhan / Graph-WaveNet

graph wavenet
MIT License
628 stars 201 forks source link

Generate_training_data #28

Closed dawn0713 closed 2 years ago

dawn0713 commented 2 years ago

Hi, I notice that in generate_training_data.py, it first uses a sliding window to reshape the training data into (num_samples - seq_len 2 + 1, seq_len, num_nodes, num_feature), then splits the training data with `num_train = round(num_samples 0.7)`, but this would cause data breach problem. Maybe it's better to split the dataset and then using a sliding window.