75df2742635b09f97e8d7ec9396750aa17046567,snips_nlu/intent_parser/builtin_intent_parser.py,BuiltinIntentParser,get_intent,#BuiltinIntentParser#Any#,45

Before Change

        if len(intents) == 0:
            return None
        else:
            return max(intents, key=lambda x: x["prob"])

    def get_entities(self, text, intent=None):
        if intent is None:
            most_likely_intent = self.get_intent(text)

After Change

        if not self.intents:
            return None

        tokenized_text = tokenize({"text": unicode(text)})
        max_proba, best_intent = -1., None
        for intent in self.intents:
            intent_classifier = IntentClassifier(
                intent_config_file=os.path.join(
                    self.configs_path, "%s.pb" % intent),
                gazetteers_dir=self.gazetteers_path
            )
            proba = intent_classifier.transform(tokenized_text)

            if (max_proba < 0.) or (proba > max_proba):
                max_proba = proba
                best_intent = intent

        return {"name": best_intent, "proba": max_proba}

    def get_entities(self, text, intent=None):
        if intent is None:
            intent = self.get_intent(text).get("name")

Instances