bug in metric calculation
This commit is contained in:
19
datasets/mask_librosa_datamodule.py
Normal file
19
datasets/mask_librosa_datamodule.py
Normal file
@@ -0,0 +1,19 @@
|
||||
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)
|
||||
Reference in New Issue
Block a user