Python MITIETrainerの例

コード例 #1

def create_trainer(config):
    backend = config.backend.lower()
    if backend == 'mitie':
        from trainers.mitie_trainer import MITIETrainer
        return MITIETrainer(config.mitie_file, config.language)
    if backend == 'spacy_sklearn':
        from trainers.spacy_sklearn_trainer import SpacySklearnTrainer
        return SpacySklearnTrainer(config, config.language)
    else:
        raise NotImplementedError("other backend trainers not implemented yet")

コード例 #2

ファイル: train.py プロジェクト: anjanisone/rasa_nlu

def create_trainer(config):
    backend = config['backend'].lower()
    if backend == 'mitie':
        from trainers.mitie_trainer import MITIETrainer
        return MITIETrainer(config['backends']['mitie'], config['language'])
    if backend == 'spacy_sklearn':
        from trainers.spacy_sklearn_trainer import SpacySklearnTrainer
        return SpacySklearnTrainer(config['backends']['spacy_sklearn'],
                                   config['language'])
    else:
        raise NotImplementedError("other backend trainers not implemented yet")