wandb를 이용해서 실험 현황을 visualization 하자

나는 train/validataion/test auc 그래프를 그리려고 사용했다.

---

사용방법

1. wandb 회원가입 후 가상환경에 설치, 로그인

1-1. wandb 홈페이지 에 접속하고 회원가입

1-2. project 생성; 노란 박스 안을 클릭해서 프로젝트를 만든다.

1-3. 프로젝트를 만들고 나면

(1) 해당 페이지가 뜨고

(2) 가상환경 activate 후 [ 1. Set ip the wandb library ] 를 따라한다. (python >= 3.7부터 가능)

$ pip install wandb

$ wandb login
을 하면 해당 메세지가 뜬다.

(3) Your API key 를 붙여 넣는다. 이때 해당 터미널엔 입력이 안되는 것처럼 보이지만 그냥 복붙 후 엔터 누르면 로그인이 된다.

• 한번 로그인하고 나면 나중엔 자동으로 됨

(4) 로그인 성공

wandb 코드

wandb를 사용하고 싶은 파일에

import wandb

wandb.login()
wandb.init(project="프로젝트 이름", entity="아이디")

아까 만든 프로젝트 이름을 넣으면 된다.

그리고 프로그램 실행하면

wandb url이 뜨는데 저기로 들어가면 된다.

나는 그냥 auc 만 비교하고 싶어서 log 기능만 사용했다.

• 중첩해서 사용

wandb.log({"CCLE CNV 698" : {'Train auc': average_auc, 'VAL AUC' : auc_val, 'TEST AUC' : auc_TEST, "val loss" : loss_val}})

• 그냥 사용

wandb.log({'VAL AUC' : auc_val,  "val loss" : loss_val})

둘의 차이는

걍 panel에 저렇게 뜬다.

wandb edit panel

edit panel을 누르면

log를 찍은 결과를 볼 수 있는데 만약 실험1, 실험2 ... 의 결과를 기록했다면 같이 볼 수 있다.
Y에서 골라주면 됨

그러면 비교해서 볼 수 있다. 👏🏻