Pseudo-MOS

import numpy as np
from versa import pseudo_mos_setup, pseudo_mos_metric

# Setup predictors
predictor_dict, predictor_fs = pseudo_mos_setup(
    predictor_types=["utmos", "dnsmos", "plcmos"],
    predictor_args={
        "dnsmos": {"fs": 16000},
        "plcmos": {"fs": 16000}
    },
    use_gpu=True
)

# Load your audio
audio = np.random.random(16000)  # Replace with actual audio
fs = 16000

# Calculate MOS scores
scores = pseudo_mos_metric(
    audio,
    fs=fs,
    predictor_dict=predictor_dict,
    predictor_fs=predictor_fs,
    use_gpu=True
)

print(f"UTMOS: {scores['utmos']:.2f}")
print(f"DNS Overall: {scores['dns_overall']:.2f}")
print(f"PLC-MOS: {scores['plcmos']:.2f}")