示例#1
0
    def test_entity_extractor_basic(self):
        x = ['what is the weather in tokyo', 'what is the weather', 'what is the weather like in kochi']
        y = [{'intent': 'weather', 'place': 'tokyo'}, {'intent': 'weather', 'place': 'here'}, {'intent': 'weather', 'place': 'kochi'}]

        ex = EntityExtractor()
        ex.fit(x, y)

        x_test = 'what is the weather in london like'
        assert ex.predict(x_test) == {'intent': 'weather', 'place': 'london'}
示例#2
0
    def test_entity_extractor_serialization(self):
        x = ['what is the weather in tokyo', 'what is the weather', 'what is the weather like in kochi']
        y = [{'intent': 'weather', 'place': 'tokyo'}, {'intent': 'weather', 'place': 'here'}, {'intent': 'weather', 'place': 'kochi'}]

        ex1 = EntityExtractor()
        ex1.fit(x, y)

        config = ex1.serialize()
        ex2 = EntityExtractor.deserialize(config)

        test_inputs = ['what is the weather in london like']

        for test_input in test_inputs:
            ex1_out = ex1.predict(test_input)
            ex2_out = ex2.predict(test_input)
            assert ex1_out == ex2_out
def test_entity_extractor_basic():
    x = [
        'what is the weather in tokyo', 'what is the weather',
        'what is the weather like in kochi'
    ]
    y = [{
        'intent': 'weather',
        'place': 'tokyo'
    }, {
        'intent': 'weather',
        'place': 'here'
    }, {
        'intent': 'weather',
        'place': 'kochi'
    }]

    ex = EntityExtractor()
    ex.fit(x, y)

    x_test = 'what is the weather like in london'
    assert ex.predict(x_test) == {'intent': 'weather', 'place': 'london'}
示例#4
0
    'what is the weather in tokyo', 'weather germany',
    'what is the weather like in kochi'
]
Y_WEATHER = [{
    'intent': 'weather',
    'place': 'tokyo'
}, {
    'intent': 'weather',
    'place': 'germany'
}, {
    'intent': 'weather',
    'place': 'kochi'
}]

EX_WEATHER = EntityExtractor()
EX_WEATHER.fit(X_WEATHER, Y_WEATHER)

X_TAXI = [
    'book a cab to kochi ', 'need a ride to delhi',
    'find me a cab for manhattan', 'call a taxi to calicut'
]
Y_TAXI = [{
    'service': 'cab',
    'destination': 'kochi'
}, {
    'service': 'ride',
    'destination': 'delhi'
}, {
    'service': 'cab',
    'destination': 'manhattan'
}, {