tommy/libpostal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[chains] chain store queries also may use street address for certain preposition types

Browse Source
This commit is contained in:
Al
2016-05-21 02:20:09 -04:00
parent 6b8e7cc2de
commit 0ab109f964
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/geodata/chains/query.py
View File

@@ -12,9 +12,9 @@ from geodata.text.normalize import normalized_tokens
from geodata.text.tokenize import tokenize, token_types
from geodata.encoding import safe_decode
ChainQuery = namedtuple('ChainQuery', 'name, prep, add_place_name')
ChainQuery = namedtuple('ChainQuery', 'name, prep, add_place_name, add_address')
NULL_CHAIN_QUERY = ChainQuery(None, None, False)
NULL_CHAIN_QUERY = ChainQuery(None, None, False, False)
class Chain(object):
@@ -94,6 +94,7 @@ class Chain(object):
prep_phrase, prep_phrase_props = weighted_choice(values, probs)
prep_phrase = safe_decode(prep_phrase)
add_address = prep_phrase_type not in (CategoryPreposition.NEARBY, CategoryPreposition.NEAR_ME, CategoryPreposition.IN)
add_place_name = prep_phrase_type not in (CategoryPreposition.NEARBY, CategoryPreposition.NEAR_ME)
return ChainQuery(chain_phrase, prep=prep_phrase, add_place_name=add_place_name)
return ChainQuery(chain_phrase, prep=prep_phrase, add_place_name=add_place_name, add_address=add_address)