Esempi in Python per Task

Esempio n. 1

File: speaker_change_detection.py Progetto: zhangpengpengpeng/pyannote-audio

    def task(self):

        if self.regression:
            return Task(type=TaskType.REGRESSION, output=TaskOutput.SEQUENCE)
        else:
            return Task(type=TaskType.MULTI_CLASS_CLASSIFICATION,
                        output=TaskOutput.SEQUENCE)

Esempio n. 2

File: resegmentation.py Progetto: zhangpengpengpeng/pyannote-audio

    def task(self):
        if self.allow_overlap:
            # when allowing overlap, multiple speakers can be active at the
            # same time (hence multi-label classification task)
            return Task(type=TaskType.MULTI_LABEL_CLASSIFICATION,
                        output=TaskOutput.SEQUENCE)

        else:
            # when overlap is not allowed, only one speaker can be active at
            # a particular time (hence: multi-class classification)
            return Task(type=TaskType.MULTI_CLASS_CLASSIFICATION,
                        output=TaskOutput.SEQUENCE)

Esempio n. 3

File: speaker_change_detection.py Progetto: zhiqizhang/pyannote-audio

    def specifications(self):
        if self.regression:
            return {
                'task': Task(type=TaskType.REGRESSION,
                             output=TaskOutput.SEQUENCE),
                'X': {'dimension': self.feature_extraction.dimension},
                'y': {'classes': ['change', ]},
            }

        else:
            return {
                'task': Task(type=TaskType.MULTI_CLASS_CLASSIFICATION,
                             output=TaskOutput.SEQUENCE),
                'X': {'dimension': self.feature_extraction.dimension},
                'y': {'classes': ['non_change', 'change']},
            }

Esempio n. 4

File: overlap_detection.py Progetto: zhiqizhang/pyannote-audio

 def specifications(self):
     return {
         'task': Task(type=TaskType.MULTI_CLASS_CLASSIFICATION,
                      output=TaskOutput.SEQUENCE),
         'X': {'dimension': self.feature_extraction.dimension},
         'y': {'classes': ['non_overlap', 'overlap']},
     }

Esempio n. 5

File: generators.py Progetto: zhiqizhang/pyannote-audio

 def specifications(self):
     return {
         'X': {
             'dimension': self.feature_extraction.dimension
         },
         'y': {
             'classes': self.segment_labels_
         },
         'task':
         Task(type=TaskType.REPRESENTATION_LEARNING,
              output=TaskOutput.VECTOR),
     }

Esempio n. 6

File: generators.py Progetto: zhangpengpengpeng/pyannote-audio

 def specifications(self):
     return {
         "X": {
             "dimension": self.feature_extraction.dimension
         },
         "y": {
             "classes": self.segment_labels_
         },
         "task":
         Task(type=TaskType.REPRESENTATION_LEARNING,
              output=TaskOutput.VECTOR),
     }

Esempio n. 7

File: domain_classification.py Progetto: zhiqizhang/pyannote-audio

 def specifications(self):
     return {
         'task':
         Task(type=TaskType.MULTI_CLASS_CLASSIFICATION,
              output=TaskOutput.VECTOR),
         'X': {
             'dimension': self.feature_extraction.dimension
         },
         'y': {
             'classes': self.file_labels_[self.domain]
         },
     }

Esempio n. 8

File: config.py Progetto: zhangpengpengpeng/pyannote-audio

def load_specs(specs_yml: Path) -> Dict:
    """

    Returns
    -------
    specs : Dict
        ['task']
        [and others]
    """

    with open(specs_yml, "r") as fp:
        specifications = yaml.load(fp, Loader=yaml.SafeLoader)
    specifications["task"] = Task.from_str(specifications["task"])
    return specifications

Esempio n. 9

File: base.py Progetto: zhiqizhang/pyannote-audio

    def specifications(self):
        """Task & sample specifications

        Returns
        -------
        specs : `dict`
            ['task'] (`pyannote.audio.train.Task`) : task
            ['X']['dimension'] (`int`) : features dimension
            ['y']['classes'] (`list`) : list of classes
        """

        specs = {
            'task': Task(type=TaskType.MULTI_CLASS_CLASSIFICATION,
                         output=TaskOutput.SEQUENCE),
            'X': {'dimension': self.feature_extraction.dimension},
            'y': {'classes': self.segment_labels_},
        }

        return specs

Esempio n. 10

 def task(self):
     return Task(
         type=TaskType.MULTI_CLASS_CLASSIFICATION, output=TaskOutput.SEQUENCE
     )

Esempio n. 11

File: base.py Progetto: zhangpengpengpeng/pyannote-audio

 def task(self):
     return Task(type=TaskType.REPRESENTATION_LEARNING,
                 output=TaskOutput.VECTOR)