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))