[openaddresses] OpenAddresses training script

scripts/geodata/openaddresses/openaddresses_training_data.py

@@ -0,0 +1,47 @@
+# -*- coding: utf-8 -*-
+'''
+openaddresses_training_data.py
+------------------------------
+
+This script generates several training sets from OpenAddresses.
+'''
+
+import argparse
+import os
+
+from geodata.openaddresses.formatter import OpenAddressesFormatter
+from geodata.polygons.language_polys import LanguagePolygonIndex
+
+
+if __name__ == '__main__':
+    # Handle argument parsing here
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument('-i', '--openaddresses-dir',
+                        help='Path to OpenAddresses directory')
+
+    parser.add_argument('-f', '--format',
+                        action='store_true',
+                        default=False,
+                        help='Save formatted addresses (slow)')
+
+    parser.add_argument('-u', '--untagged',
+                        action='store_true',
+                        default=False,
+                        help='Save untagged formatted addresses (slow)')
+
+    parser.add_argument('--language-rtree-dir',
+                        required=True,
+                        help='Language RTree directory')
+
+    parser.add_argument('-o', '--out-dir',
+                        default=os.getcwd(),
+                        help='Output directory')
+
+    args = parser.parse_args()
+
+    language_rtree = LanguagePolygonIndex.load(args.language_rtree_dir)
+
+    if args.openaddresses_dir and args.format:
+        oa_formatter = OpenAddressesFormatter(language_rtree)
+        oa_formatter.build_training_data(args.openaddresses_dir, args.out_dir, tag_components=not args.untagged)