| from torch.utils.data import Dataset | |
| class SOMOSDataset(Dataset): | |
| def __init__(self, data_path: str, metric: str) -> None: | |
| """ | |
| metric is one of ['sbs', 'mos'] | |
| """ | |
| self.metric = metric | |
| class StdSOMOSDataset(SOMOSDataset): | |
| """ | |
| Standartized by user SOMOS dataset | |
| """ | |
| pass |