import torch
from torchmetrics.nominal import CramersV
metric = CramersV(num_classes=5)
metric.update(torch.randint(0, 4, (100,)), torch.randint(0, 4, (100,)))
fig_, ax_ = metric.plot()