1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
| import pynini
| from fun_text_processing.inverse_text_normalization.zh.graph_utils import (
| DAMO_SPACE,
| DAMO_NOT_QUOTE,
| GraphFst,
| delete_space,
| )
| from pynini.lib import pynutil
|
|
| class CardinalFst(GraphFst):
| """
| Finite state transducer for verbalizing cardinal
| e.g. cardinal { integer: "23" negative: "-" } -> -23
| """
|
| def __init__(self):
| super().__init__(name="cardinal", kind="verbalize")
| optional_sign = pynini.closure(
| pynutil.delete("negative:")
| + delete_space
| + pynutil.delete('"')
| + DAMO_NOT_QUOTE
| + pynutil.delete('"'),
| 0,
| 1,
| )
| graph = (
| pynutil.delete("integer:")
| + delete_space
| + pynutil.delete('"')
| + pynini.closure(DAMO_NOT_QUOTE, 1)
| + pynutil.delete('"')
| )
| self.numbers = graph
| graph = optional_sign + graph
| delete_tokens = self.delete_tokens(graph)
| self.fst = delete_tokens.optimize()
|
|
