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

[osm] Omitting country in limited address data set (often abbreviated, doesn't convey language as well)

Browse Source
This commit is contained in:
Al
2015-08-15 03:25:45 -04:00
parent c505260912
commit 89071ea21a
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
scripts/geodata/osm/osm_address_training_data.py
View File

@@ -448,9 +448,14 @@ NAME_KEYS = (
'name',
'addr:housename',
)
COUNTRY_KEYS = (
'country',
'country_name',
'addr:country',
)
def build_address_format_training_data_sans_names(language_rtree, infile, out_dir):
def build_address_format_training_data_limited(language_rtree, infile, out_dir):
i = 0
formatter = AddressFormatter()
@@ -468,7 +473,7 @@ def build_address_format_training_data_sans_names(language_rtree, infile, out_di
if not (country and default_languages):
continue
for key in NAME_KEYS:
for key in NAME_KEYS + COUNTRY_KEYS:
_ = value.pop(key, None)
if not value:
@@ -563,10 +568,10 @@ if __name__ == '__main__':
default=False,
help='Save formatted addresses (slow)')
parser.add_argument('-n', '--no-house-names',
parser.add_argument('-l', '--limited-addresses',
action='store_true',
default=False,
help='Save formatted addresses without house names (slow)')
help='Save formatted addresses without house names or country (slow)')
parser.add_argument('-t', '--temp-dir',
default=tempfile.gettempdir(),
@@ -589,11 +594,11 @@ if __name__ == '__main__':
# Can parallelize
if args.streets_file:
build_ways_training_data(language_rtree, args.streets_file, args.out_dir)
if args.address_file and not args.format_only and not args.no_house_names:
if args.address_file and not args.format_only and not args.limited_addresses:
build_address_training_data(language_rtree, args.address_file, args.out_dir)
if args.address_file and args.format_only:
build_address_format_training_data(language_rtree, args.address_file, args.out_dir)
if args.address_file and args.no_house_names:
build_address_format_training_data_sans_names(language_rtree, args.address_file, args.out_dir)
if args.address_file and args.limited_addresses:
build_address_format_training_data_limited(language_rtree, args.address_file, args.out_dir)
if args.venues_file:
build_venue_training_data(language_rtree, args.venues_file, args.out_dir)