HAICon2021 산업제어시스템 보안위협 탐지 AI 경진대회

알고리즘 | 정형 | 이상탐지 | 제어 | TaPR

Prize : 총 2,000만원
903명 마감

Practice

Overview Data Code (share) Talk Leaderboard

Submission

TF(2.5+) Transformer(Encoder) Baseline/TFRecord

Jamm

2021.09.10 16:39 4,317 Views language

저는 파이토치를 거의 못쓰기때문에... 텐서플로우로 베이스라인을 작성해봤습니다. 코드를 돌리면서 작성하느라 캐글 노트북 환경을 사용했습니다. (TF 2.4.1)

train 데이터 샘플수가 100만건이나 되기 때문에 timestep 을 길게 잡고 sequence 데이터로 변형하려다 보면 메모리가 부족하다고 컴퓨터가 뻗어버리거나, 한참동안 사경을 헤매다가 돌아오기 때문에 그를 해결하기위하여 학습데이터를 TFRecord 파일로 작성했습니다.

(MultiHeadAttention 레이어를 쓰기 위해서는 TF 2.4 이상이 필요합니다!)

대회문의에서 텐서플로우 베이스라인은 없는가 에 관하여 문의가 있었던 것으로 기억하는데.. 텐서플로우 사용자분들께 도움이 되었으면 좋겠습니다.
도움이 되셨다면 좋아요와 댓글...구독(?)과 알림설정(?) 부탁드립니다! 감사합니다.

Code