def __init__(self, factory_config):
super(BuiltinEntityMatchFactory, self).__init__(factory_config)
self.tagging_scheme = TaggingScheme(self.args["tagging_scheme_code"])
self.builtin_entities = None
self.builtin_entities = self.args.get("entity_labels")
self._language = None
self.language = self.args.get("language_code")
def __init__(self, factory_config):
super(EntityMatchFactory, self).__init__(factory_config)
self.use_stemming = self.args["use_stemming"]
self.tagging_scheme = TaggingScheme(self.args["tagging_scheme_code"])
self.collections = self.args.get("collections")
self._language = None
self.language = self.args.get("language_code")
def __init__(self, factory_config, **shared):
super(CustomEntityMatchFactory, self).__init__(factory_config,
**shared)
self.use_stemming = self.args["use_stemming"]
self.tagging_scheme = TaggingScheme(self.args["tagging_scheme_code"])
self._entities = None
self.entities = self.args.get("entities")
def tagging_scheme(self, value):
from snips_nlu.slot_filler.crf_utils import TaggingScheme
if isinstance(value, TaggingScheme):
self._tagging_scheme = value
elif isinstance(value, int):
self._tagging_scheme = TaggingScheme(value)
else:
raise TypeError("Expected instance of TaggingScheme or int but"
"received: %s" % type(value))
