use better logging and provide --verbose option in NER_plaintexts.py

MiMoText · Nov 25, 2021 · 971dba8 · 971dba8
1 parent 54a4a45
commit 971dba8
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/NER-with-SpaCy/scripts/NER_plaintexts.py b/NER-with-SpaCy/scripts/NER_plaintexts.py
@@ -209,8 +209,16 @@ def main(args):
         type=int,
         help='the number of most frequent NEs to store'
         )
+    arg_parser.add_argument(
+        '-v', '--verbose',
+        action='store_true',
+        help='wether to use more verbose logging output')
 
     args = arg_parser.parse_args()
+
+    loglevel = logging.DEBUG if args.verbose else logging.WARNING
+    logging.basicConfig(format='%(asctime)s -- %(message)s', datefmt='%Y-%m-%d %I:%M:%S', level=loglevel)
+
     logging.info('running with the following options')
     logging.info(args)
     main(args)