def __init__(self, component_config=None):
if not component_config:
component_config = {}
# makes sure the name of the configuration is part of the config
# this is important for e.g. persistence
component_config["name"] = self.name
self.component_config = config.override_defaults(
self.defaults, component_config)
self.partial_processing_pipeline = None
self.partial_processing_context = None
def __init__(self, component_config=None):
if not component_config:
component_config = {}
# makes sure the name of the configuration is part of the config
# this is important for e.g. persistence
component_config["name"] = self.name
self.component_config = config.override_defaults(
self.defaults, component_config)
self.partial_processing_pipeline = None
self.partial_processing_context = None
def create(cls,
component_config: Dict[Text, Any],
config: RasaNLUModelConfig) -> 'MitieNLP':
import mitie
component_config = override_defaults(cls.defaults, component_config)
model_file = component_config.get("model")
if not model_file:
raise Exception("The MITIE component 'MitieNLP' needs "
"the configuration value for 'model'."
"Please take a look at the "
"documentation in the pipeline section "
"to get more info about this "
"parameter.")
extractor = mitie.total_word_feature_extractor(model_file)
cls.ensure_proper_language_model(extractor)
return cls(component_config, extractor)
def create(cls,
component_config: Dict[Text, Any],
config: RasaNLUModelConfig) -> 'SpacyNLP':
import spacy
component_config = override_defaults(cls.defaults, component_config)
spacy_model_name = component_config.get("model")
# if no model is specified, we fall back to the language string
if not spacy_model_name:
spacy_model_name = config.language
component_config["model"] = config.language
logger.info("Trying to load spacy model with "
"name '{}'".format(spacy_model_name))
nlp = spacy.load(spacy_model_name, disable=['parser'])
cls.ensure_proper_language_model(nlp)
return cls(component_config, nlp)
def for_component(self, name, defaults=None):
for c in self.get('pipeline', []):
if c.get("name") == name:
return override_defaults(defaults, c)
else:
return defaults or {}
def __init__(self, component_config=None):
self.component_config = config.override_defaults(
self.defaults, component_config)
self.partial_processing_pipeline = None
self.partial_processing_context = None
def for_component(self, name, defaults=None):
for c in self.get('pipeline', []):
if c.get("name") == name:
return override_defaults(defaults, c)
else:
return defaults or {}
def __init__(self, component_config=None):
self.component_config = config.override_defaults(
self.defaults, component_config)
self.partial_processing_pipeline = None
self.partial_processing_context = None
