[머신 러닝] 텐서보드를 이용해 시각화하기

텐서보드를 이용해 시각화하기

텐서보드를 이용해 훈련하는 동안 학습 곡선을 그리거나 여러 실행 간의 학습 곡선을 비교하고 계산 그래프 시각화와 훈련 통계 분석을 수행할 수 있습니다. 또한 모델이 생성한 이미지를 확인하거나 3D에 투영된 복잡한 다차원 데이터를 시각화하고 자동으로 클러스터링을 해주는 등 많은 기능을 제공합니다.

 

텐서보드를 이용하려면 프로그램을 수정하여 이벤트 파일이라는 이진 로그 파일에 시각화하려는 데이터를 출력해야합니다. 각각의 이진 데이터 레코드를 서머리라고 합니다. 텐서보드 서버는 로그 디렉터리를 모니터링하고 자동으로 변경사항을 읽어 그래프를 업데이트합니다. 훈련하는 중간에 학습 곡선같이 실시간 데이터를 시각화할 수 있습니다.

 

일반적으로 텐서보드 서버가 루트 로그 디렉터리를 가리키고

프로그램은 실행 시마다 다른 서브 디렉터리에 이벤트를 기록합니다.

이렇게 하고 났다면 텐서보드 서버를 시작해야 합니다.

터미널 명령으로 아래와 같이 실행할 수도 있고,

$ tensorboard --logdir=./my_logs --port=6006

주피터 노트북에서 바로 아래와 같이 실행할 수도 있습니다.

%load_ext tensorboard  # 텐서보드 확장 로드
%tensorboard --logdir=./my_logs --port=6006  # 포트 6006에서 텐서보드 서버 실행 후 접속

그럼 이와 같은 화면을 볼 수 있습니다.

텐서보드