Repository navigation
Expand file tree
/
Copy path__init__.py
More file actions
53 lines (48 loc) · 2.46 KB
/
Copy path__init__.py
File metadata and controls
53 lines (48 loc) · 2.46 KB
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
# -*- coding: utf-8 -*-
from .parsers import (BiaffineDependencyParser,
BiaffineSemanticDependencyParser, CRF2oDependencyParser,
CRFConstituencyParser, CRFDependencyParser, Parser,
VIConstituencyParser, VIDependencyParser,
VISemanticDependencyParser)
__all__ = ['BiaffineDependencyParser',
'CRFDependencyParser',
'CRF2oDependencyParser',
'VIDependencyParser',
'CRFConstituencyParser',
'VIConstituencyParser',
'BiaffineSemanticDependencyParser',
'VISemanticDependencyParser',
'Parser']
__version__ = '1.1.2'
PARSER = {parser.NAME: parser for parser in [BiaffineDependencyParser,
CRFDependencyParser,
CRF2oDependencyParser,
VIDependencyParser,
CRFConstituencyParser,
VIConstituencyParser,
BiaffineSemanticDependencyParser,
VISemanticDependencyParser]}
SRC = {'github': 'https://github.com/yzhangcs/parser/releases/download',
'hlt': 'http://hlt.suda.edu.cn/~yzhang/supar'}
NAME = {
'biaffine-dep-en': 'ptb.biaffine.dep.lstm.char',
'biaffine-dep-zh': 'ctb7.biaffine.dep.lstm.char',
'crf2o-dep-en': 'ptb.crf2o.dep.lstm.char',
'crf2o-dep-zh': 'ctb7.crf2o.dep.lstm.char',
'biaffine-dep-roberta-en': 'ptb.biaffine.dep.roberta',
'biaffine-dep-electra-zh': 'ctb7.biaffine.dep.electra',
'biaffine-dep-xlmr': 'ud.biaffine.dep.xlmr',
'crf-con-en': 'ptb.crf.con.lstm.char',
'crf-con-zh': 'ctb7.crf.con.lstm.char',
'crf-con-roberta-en': 'ptb.crf.con.roberta',
'crf-con-electra-zh': 'ctb7.crf.con.electra',
'crf-con-xlmr': 'spmrl.crf.con.xlmr',
'biaffine-sdp-en': 'dm.biaffine.sdp.lstm.tag-char-lemma',
'biaffine-sdp-zh': 'semeval16.biaffine.sdp.lstm.tag-char-lemma',
'vi-sdp-en': 'dm.vi.sdp.lstm.tag-char-lemma',
'vi-sdp-zh': 'semeval16.vi.sdp.lstm.tag-char-lemma',
'vi-sdp-roberta-en': 'dm.vi.sdp.roberta',
'vi-sdp-electra-zh': 'semeval16.vi.sdp.electra'
}
MODEL = {src: {n: f"{link}/v1.1.0/{m}.zip" for n, m in NAME.items()} for src, link in SRC.items()}
CONFIG = {src: {n: f"{link}/v1.1.0/{m}.ini" for n, m in NAME.items()} for src, link in SRC.items()}