from torch import randint
from torchmetrics.classification import MulticlassFBetaScore
metric = MulticlassFBetaScore(num_classes=3, beta=2.0, average=None)
metric.update(randint(3, (20,)), randint(3, (20,)))
fig_, ax_ = metric.plot()