import wandb
# 새로운 run 초기화
with wandb.init(project="visualize-predictions", name="tables") as run:
# 리스트의 리스트를 사용하여 테이블 데이터 생성
data = [["Cat", "1", "1"],["Dog", "0", "-1"]]
run.log({"Table 1": wandb.Table(data=data, columns=["Text", "Predicted Label", "True Label"])})
# `wandb.Table.add_data()` 메소드를 사용하여 테이블 데이터 생성
table = wandb.Table(columns=["Text", "Predicted Label", "True Label"])
table.add_data("Cat", "1", "1")
table.add_data("Dog", "0", "-1")
run.log({"Table 2": table})