20 lines
642 B
Python
20 lines
642 B
Python
from datasets.compare_base import CompareBase
|
|
from ml_lib.utils.tools import add_argparse_args
|
|
|
|
|
|
class MaskLibrosaDatamodule(CompareBase):
|
|
|
|
class_names = ['mask', 'clear']
|
|
sub_dataset_name = 'ComParE2020_Mask'
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super(MaskLibrosaDatamodule, self).__init__(*args, **kwargs)
|
|
|
|
@classmethod
|
|
def add_argparse_args(cls, parent_parser):
|
|
return add_argparse_args(CompareBase, parent_parser)
|
|
|
|
@classmethod
|
|
def from_argparse_args(cls, args, **kwargs):
|
|
return CompareBase.from_argparse_args(args, class_names=cls.class_names, sub_dataset_name=cls.sub_dataset_name)
|