7.4.7 서브미션 노트북 만들기

Jigsaw 대회는 Code Competition이므로 결과를 제출하려면 서브미션 노트북이 별개로 필요합니다. 제출을 위해 노트북을 새로 만들고 이전 단계에서 학습된 모델을 불러와 주어진 테스트 셋에 대한 정답을 추론하는 코드를 만들 것입니다.

먼저 그림 7-18과 같이 Jigsaw 데이터와 이제껏 만들었던 TFRecord 데이터셋, 그리고 TPU로 학습시킨 BERT 모델의 가중치를 불러옵니다. 학습한 BERT 모델의 가중치는 앞서 TPU 학습에 사용한 캐글 노트북의 Output을 캐글 데이터셋으로 먼저 생성한 뒤 이를 추가하여 사용할 수 있습니다.

▲ 그림 7-18 서브미션 노트북에 데이터 추가