Compute confidence scores by char, word or line

Closes #33 (closed)

dan/predict/__init__.py

@@ -61,6 +61,14 @@ def add_predict_parser(subcommands) -> None:
         help="Whether to return confidence scores.",
         required=False,
     )
+    parser.add_argument(
+        "--confidence-score-levels",
+        default="",
+        type=str,
+        nargs="+",
+        help="Levels of confidence scores. Should be a list of any combinaison of ['char', 'word', 'line'].",
+        required=False,
+    )
     parser.add_argument(
         "--attention-map",
         action="store_true",
@@ -82,5 +90,20 @@ def add_predict_parser(subcommands) -> None:
         help="Image scaling factor before creating the GIF",
         required=False,
     )
-
+    parser.add_argument(
+        "--word-separators",
+        default=[" ", "\n"],
+        type=str,
+        nargs="+",
+        help="String separators used to split text into words.",
+        required=False,
+    )
+    parser.add_argument(
+        "--line-separators",
+        default=["\n"],
+        type=str,
+        nargs="+",
+        help="String separators used to split text into lines.",
+        required=False,
+    )
     parser.set_defaults(func=run)