import torch from torchmetrics.audio import ShortTimeObjectiveIntelligibility metric = ShortTimeObjectiveIntelligibility(8000, False) g = torch.manual_seed(1) preds = torch.randn(8000) target = torch.randn(8000) values = [ ] for _ in range(10): values.append(metric(preds, target)) fig_, ax_ = metric.plot(values)