def test_should_be_serializable_into_bytearray(self):
# Given
dataset_stream = io.StringIO("""
---
type: intent
name: MakeTea
utterances:
- make me a cup of tea
- i want two cups of tea please
- can you prepare one cup of tea ?
---
type: intent
name: MakeCoffee
utterances:
- make me a cup of coffee please
- brew two cups of coffee
- can you prepare one cup of coffee""")
dataset = Dataset.from_yaml_files("en", [dataset_stream]).json
shared = self.get_shared_data(dataset)
intent_classifier = LogRegIntentClassifier(**shared).fit(dataset)
# When
intent_classifier_bytes = intent_classifier.to_byte_array()
loaded_classifier = LogRegIntentClassifier.from_byte_array(
intent_classifier_bytes, **shared)
result = loaded_classifier.get_intent("make me two cups of tea")
# Then
expected_intent = "MakeTea"
self.assertEqual(expected_intent, result[RES_INTENT_NAME])
def test_should_be_serializable_into_bytearray(self):
# Given
dataset = BEVERAGE_DATASET
intent_classifier = LogRegIntentClassifier().fit(dataset)
# When
intent_classifier_bytes = intent_classifier.to_byte_array()
loaded_classifier = LogRegIntentClassifier.from_byte_array(
intent_classifier_bytes)
result = loaded_classifier.get_intent("make me two cups of tea")
# Then
expected_intent = "MakeTea"
self.assertEqual(expected_intent, result[RES_INTENT_NAME])
def test_should_be_serializable_into_bytearray(self):
# Given
dataset = validate_and_format_dataset(BEVERAGE_DATASET)
intent_classifier = LogRegIntentClassifier().fit(dataset)
# When
intent_classifier_bytes = intent_classifier.to_byte_array()
custom_entity_parser = CustomEntityParser.build(
dataset, CustomEntityParserUsage.WITHOUT_STEMS)
builtin_entity_parser = BuiltinEntityParser.build(language="en")
loaded_classifier = LogRegIntentClassifier.from_byte_array(
intent_classifier_bytes,
builtin_entity_parser=builtin_entity_parser,
custom_entity_parser=custom_entity_parser)
result = loaded_classifier.get_intent("make me two cups of tea")
# Then
expected_intent = "MakeTea"
self.assertEqual(expected_intent, result[RES_INTENT_NAME])
