Minor modifications

yzhangcs · yzhangcs · commit 2548c2402f8d · 2020-11-27T13:07:53.000+08:00
diff --git a/supar/cmds/biaffine_dependency.py b/supar/cmds/biaffine_dependency.py
@@ -13,6 +13,7 @@ def main():
     parser.add_argument('--partial', action='store_true', help='whether partial annotation is included')
     parser.set_defaults(Parser=BiaffineDependencyParser)
     subparsers = parser.add_subparsers(title='Commands', dest='mode')
+    # train
     subparser = subparsers.add_parser('train', help='Train a parser.')
     subparser.add_argument('--feat', '-f', choices=['tag', 'char', 'bert'], help='choices of additional features')
     subparser.add_argument('--build', '-b', action='store_true', help='whether to build the model first')
diff --git a/supar/cmds/crf2o_dependency.py b/supar/cmds/crf2o_dependency.py
@@ -14,6 +14,7 @@ def main():
     parser.add_argument('--proj', action='store_true', help='whether to projectivize the data')
     parser.add_argument('--partial', action='store_true', help='whether partial annotation is included')
     subparsers = parser.add_subparsers(title='Commands', dest='mode')
+    # train
     subparser = subparsers.add_parser('train', help='Train a parser.')
     subparser.add_argument('--feat', '-f', choices=['tag', 'char', 'bert'], help='choices of additional features')
     subparser.add_argument('--build', '-b', action='store_true', help='whether to build the model first')
diff --git a/supar/cmds/crf_constituency.py b/supar/cmds/crf_constituency.py
@@ -11,6 +11,7 @@ def main():
     parser.set_defaults(Parser=CRFConstituencyParser)
     parser.add_argument('--mbr', action='store_true', help='whether to use MBR decoding')
     subparsers = parser.add_subparsers(title='Commands', dest='mode')
+    # train
     subparser = subparsers.add_parser('train', help='Train a parser.')
     subparser.add_argument('--feat', '-f', choices=['tag', 'char', 'bert'], help='choices of additional features')
     subparser.add_argument('--build', '-b', action='store_true', help='whether to build the model first')
diff --git a/supar/cmds/crf_dependency.py b/supar/cmds/crf_dependency.py
@@ -14,6 +14,7 @@ def main():
     parser.add_argument('--proj', action='store_true', help='whether to projectivize the data')
     parser.add_argument('--partial', action='store_true', help='whether partial annotation is included')
     subparsers = parser.add_subparsers(title='Commands', dest='mode')
+    # train
     subparser = subparsers.add_parser('train', help='Train a parser.')
     subparser.add_argument('--feat', '-f', choices=['tag', 'char', 'bert'], help='choices of additional features')
     subparser.add_argument('--build', '-b', action='store_true', help='whether to build the model first')
diff --git a/supar/cmds/crfnp_dependency.py b/supar/cmds/crfnp_dependency.py
@@ -11,6 +11,7 @@ def main():
     parser.set_defaults(Parser=CRFNPDependencyParser)
     parser.add_argument('--mbr', action='store_true', help='whether to use MBR decoding')
     subparsers = parser.add_subparsers(title='Commands', dest='mode')
+    # train
     subparser = subparsers.add_parser('train', help='Train a parser.')
     subparser.add_argument('--feat', '-f', choices=['tag', 'char', 'bert'], help='choices of additional features')
     subparser.add_argument('--build', '-b', action='store_true', help='whether to build the model first')
diff --git a/supar/utils/metric.py b/supar/utils/metric.py
@@ -24,7 +24,7 @@ def score(self):
 
 class AttachmentMetric(Metric):
 
-    def __init__(self, eps=1e-8):
+    def __init__(self, eps=1e-12):
         super().__init__()
 
         self.eps = eps
@@ -79,7 +79,7 @@ def las(self):
 
 class SpanMetric(Metric):
 
-    def __init__(self, eps=1e-8):
+    def __init__(self, eps=1e-12):
         super().__init__()
 
         self.n = 0.0
@@ -154,7 +154,7 @@ def lf(self):
 
 class ChartMetric(Metric):
 
-    def __init__(self, eps=1e-5):
+    def __init__(self, eps=1e-12):
         super(ChartMetric, self).__init__()
 
         self.tp = 0.0
