def test_should_get_intents(self):
# Given
dataset_stream = io.StringIO("""
---
type: intent
name: intent1
utterances:
- yala yili
---
type: intent
name: intent2
utterances:
- yala yili yulu
---
type: intent
name: intent3
utterances:
- yili yulu yele""")
dataset = Dataset.from_yaml_files("en", [dataset_stream]).json
shared = self.get_shared_data(dataset)
shared[RANDOM_STATE] = 42
parser = ProbabilisticIntentParser(**shared).fit(dataset)
text = "yala yili yulu"
# When
results = parser.get_intents(text)
intents = [res[RES_INTENT_NAME] for res in results]
# Then
expected_intents = ["intent2", "intent1", "intent3", None]
self.assertEqual(expected_intents, intents)
def test_should_get_intents(self):
# Given
dataset_stream = io.StringIO("""
---
type: intent
name: intent1
utterances:
- yala yili
---
type: intent
name: intent2
utterances:
- yala yili yulu
---
type: intent
name: intent3
utterances:
- yili yulu yele""")
dataset = Dataset.from_yaml_files("en", [dataset_stream]).json
classifier_config = LogRegIntentClassifierConfig(random_seed=42)
parser_config = ProbabilisticIntentParserConfig(classifier_config)
parser = ProbabilisticIntentParser(parser_config).fit(dataset)
text = "yala yili yulu"
# When
results = parser.get_intents(text)
intents = [res[RES_INTENT_NAME] for res in results]
# Then
expected_intents = ["intent2", "intent1", "intent3", None]
self.assertEqual(expected_intents, intents)