# -*- coding: utf-8 -*-
# @Author: colinzhang
# @Date:   9/11/19 12:59 PM

discourse_tagset = {
    '0': 1,
    'Expansion': 2,
    'Contingency': 3,
    'Comparison': 4,
    'Temporal': 5
}

modal_tagset = {
    '0': 1,
    'MD': 2
}

wn_pos_tag_tagset = {
    "": 1,
    "a": 2,
    "n": 3,
    "r": 4,
    "v": 5
}

pos_tag_tagset = {
    "$": 1,
    "''": 2,
    "(": 3,
    ")": 4,
    ",": 5,
    "--": 6,
    ".": 7,
    ":": 8,
    "CC": 9,
    "CD": 10,
    "DT": 11,
    "EX": 12,
    "FW": 13,
    "IN": 14,
    "JJ": 15,
    "JJR": 16,
    "JJS": 17,
    "LS": 18,
    "MD": 19,
    "NN": 20,
    "NNP": 21,
    "NNPS": 22,
    "NNS": 23,
    "PDT": 24,
    "POS": 25,
    "PRP": 26,
    "PRP$": 27,
    "RB": 28,
    "RBR": 29,
    "RBS": 30,
    "RP": 31,
    "SYM": 32,
    "TO": 33,
    "UH": 34,
    "VB": 35,
    "VBD": 36,
    "VBG": 37,
    "VBN": 38,
    "VBP": 39,
    "VBZ": 40,
    "WDT": 41,
    "WP": 42,
    "WP$": 43,
    "WRB": 44,
    "``": 45,
    "#": 46
}

argumentation_tagset = {
    "O": 1,
    "MC-B": 2,
    "MC-I": 3,
    "C-B": 4,
    "C-I": 5,
    "P-B": 6,
    "P-I": 7,
    '-X-': 8
}

sent_function_label_tagset = {
    "Elaboration": 1,
    "Prompt": 2,
    "Transition": 3,
    "Thesis": 4,
    "MainIdea": 5,
    "Support": 6,
    "Conclusion": 7,
    "Rebuttal": 8,
    "Solution": 9,
    "Suggestion": 10
}
