본문 바로가기

Data Science/Time series data

[ML] Informer 사용

728x90

https://github.com/zhouhaoyi/Informer2020

 

zhouhaoyi/Informer2020

The GitHub repository for the paper "Informer" accepted by AAAI 2021. - zhouhaoyi/Informer2020

github.com

 

 

 

Colab 코드 사용

 

코드 사용 주의사항

1) custom 데이터셋 적용

- custom 데이터셋을 사용하고 싶을 때는 args.data = "custom"

- 이러면 args에 입력한 대로 custom dataset을 자동으로 만들어서 적용시킴

 

2) timestamp column

- timestamp column의 이름은 date 이어야 하며

- timestamp column이 가장 첫번째 column이 어야 한다

 

이해 필요

1) label_seq의 역할이 무엇인가

- label_seq : start token length of Informer decoder 

- decoder의 start token은 무엇인가

 

2) time features의 역할

 

 

개선 사항

1) train/val/test 비율 조정 가능하도록 수정 (data_loader.py 참고) 

- train:0.7, val:0.1, test:0.2로 고정되어 있음

 

2) Scaling

- 자체적으로 만든 Standard Scaler를 사용하며, 다른 scaling을 사용하고 싶을 때는 코드를 수정해야함 (data_loader.py 참고) 

 

반응형