File size: 914 Bytes
fed5c73
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
from enum import Enum


class DucklingLocaleTypes(Enum):
    EN = "en_EN"
    AR = "ar_AR"


class DucklingDimensionTypes(Enum):
    AMOUNT_OF_MONEY = "amount-of-money"
    NUMERAL = "numeral"
    ORDINAL = "ordinal"
    VOLUME = "volume"
    QUANTITY = "quantity"


class ModelTypes(Enum):
    DUCKLING = "duckling"
    DUCKLING_OPERATOR = "duckling_operator"
    T5_NER = "t5_ner"


class OperatorModelTypes(Enum):
    FASTTEXT = "fasttext"
    BERT = "bert"


class OperatorClassTypes(Enum):
    EQ = "__label__EQ"
    LE = "__label__LE"
    GE = "__label__GE"
    GELE = "__label__GELE"
    NONE = "__label__NONE"


class T5DomainClassTypes(Enum):
    RATE = "[RATE]"
    SPECS = "[SPECS]"
    BRAND = "[BRAND]"
    SUBCATEGORY = "[SUBCATEGORY]"
    SUPERCATEGORY = "[SUPERCATEGORY]"
    PRICE = "[PRICE]"


class T5PriceSubclassTypes(Enum):
    EQ = "EQ"
    LE = "LE"
    GE = "GE"
    CURRENCY = "CURRENCY"