import wandb
# Initialize a new run
with wandb.init(project="visualize-predictions", name="tables") as run:
# Create tabular data, using a list of lists
data = [["Cat", "1", "1"],["Dog", "0", "-1"]]
run.log({"Table 1": wandb.Table(data=data, columns=["Text", "Predicted Label", "True Label"])})
# Create tabular data, using `wandb.Table.add_data()` method
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})