for each turbine, we have 3 models:
1 for lightgbm(namely lgn)
1 for first 144 timestamps prediction(namely GRU-FH)
1 for 288 timestamps prediction(namely GRU-ALL).
more information about KDDCUP2022
before training, you need to upload the dataset into folder LGB_train and GRU_train.
- modify the file prepare.py to your path.
- data proprecess
python datapreprocess.py #Bash
- LGB Train
python train_split_smooth.py #Bash
- modify the file prepare.py to your path.
- if you want to train GRU-ALL, set the output_len 288,
elif you want to train GRU-FH, set the output_len 144. - model training
python train.py #Bash
before testing, put the pretrained model into folder checkpoints of prediction
- modify the file prepare.py and predict.py to your own model path.
- submit the prediction folder for online test.
Any problems please contact me at jackie64321@gmail.com