@@ -46,39 +46,37 @@ def __init__(self,
4646 self .build_fst ('zh_itn' , cache_dir , overwrite_cache )
4747
4848 def build_tagger (self ):
49- tagger = (
50- add_weight (Date ().tagger , 1.02 )
51- | add_weight (Whitelist ().tagger , 1.01 )
52- | add_weight (Fraction ().tagger , 1.05 )
53- | add_weight (
54- Measure (enable_0_to_9 = self .enable_0_to_9 ).tagger , 1.05 ) # noqa
55- | add_weight (Money (enable_0_to_9 = self .enable_0_to_9 ).tagger ,
56- 1.04 ) # noqa
57- | add_weight (Time ().tagger , 1.05 )
58- | add_weight (
59- Cardinal (self .convert_number , self .enable_0_to_9 ,
60- self .enable_million ).tagger , 1.06 ) # noqa
61- | add_weight (Math ().tagger , 1.10 )
62- | add_weight (LicensePlate ().tagger , 1.0 )
63- | add_weight (Char ().tagger , 100 )).optimize ()
49+ tagger = (add_weight (Date ().tagger , 1.02 )
50+ | add_weight (Whitelist ().tagger , 1.01 )
51+ | add_weight (Fraction ().tagger , 1.05 )
52+ | add_weight (
53+ Measure (enable_0_to_9 = self .enable_0_to_9 ).tagger , 1.05 )
54+ | add_weight (
55+ Money (enable_0_to_9 = self .enable_0_to_9 ).tagger , 1.04 )
56+ | add_weight (Time ().tagger , 1.05 )
57+ | add_weight (
58+ Cardinal (self .convert_number , self .enable_0_to_9 ,
59+ self .enable_million ).tagger , 1.06 )
60+ | add_weight (Math ().tagger , 1.10 )
61+ | add_weight (LicensePlate ().tagger , 1.0 )
62+ | add_weight (Char ().tagger , 100 )).optimize ()
6463
6564 tagger = tagger .star
6665 # remove the last space
6766 self .tagger = tagger @ self .build_rule (delete (' ' ), '' , '[EOS]' )
6867
6968 def build_verbalizer (self ):
70- verbalizer = (
71- Cardinal (self .convert_number , self .enable_0_to_9 ,
72- self .enable_million ).verbalizer # noqa
73- | Char ().verbalizer
74- | Date ().verbalizer
75- | Fraction ().verbalizer
76- | Math ().verbalizer
77- | Measure (enable_0_to_9 = self .enable_0_to_9 ).verbalizer
78- | Money (enable_0_to_9 = self .enable_0_to_9 ).verbalizer
79- | Time ().verbalizer
80- | LicensePlate ().verbalizer
81- | Whitelist ().verbalizer ).optimize ()
69+ verbalizer = (Cardinal (self .convert_number , self .enable_0_to_9 ,
70+ self .enable_million ).verbalizer
71+ | Char ().verbalizer
72+ | Date ().verbalizer
73+ | Fraction ().verbalizer
74+ | Math ().verbalizer
75+ | Measure (enable_0_to_9 = self .enable_0_to_9 ).verbalizer
76+ | Money (enable_0_to_9 = self .enable_0_to_9 ).verbalizer
77+ | Time ().verbalizer
78+ | LicensePlate ().verbalizer
79+ | Whitelist ().verbalizer ).optimize ()
8280 postprocessor = PostProcessor (remove_interjections = True ).processor
8381
8482 self .verbalizer = (verbalizer @ postprocessor ).star
0 commit comments