raven_utils/constant.py

RAVEN = "arr"
RAVEN_BIG = "arrb"
INDEX = "index"
LABELS = "labels"
TARGET_LABELS = "target_labels"
FEATURES = "features"
ACC_SAME = "acc_same"
ACC_CHOOSE_LOWER = "acc_choose_lower"
ACC_CHOOSE_UPPER = "acc_choose_upper"
ACC_NO_GROUP = "acc_NO_group"
CLASSIFICATION = "classification"
INFERENCE = "inference"
# PROPERTIES = "properties"
PROPERTY = "property"
MEMORY = "memory"
CONTROL = "control"
LATENT = "latent"
TARGET = "target"
INPUTS = "inputs"
RES = "res"
RESULT = "result"
MERGE = "merge"
MEMORY_STATE = "memory_state"
CONTROL_STATE = "control_state"
CONCAT = "concat"
FLATTEN = "flatten"
CROSS_ENTROPY = "cross_entropy"
SLOT = "slot"
PROPERTIES = "properties"
ACC = "acc"
GROUP = 'group'
NUMBER = 'number'
TRANS = 'trans'
TAIL = "tail"
MASK = "mask"

RAV_METRICS = [
    ACC_NO_GROUP,
    ACC_SAME,
    ACC_CHOOSE_UPPER,
    ACC_CHOOSE_LOWER,
    "acc",
    "c_acc_NO_group",
    "c_acc",
    "loss",
]

IMP_RAV_METRICS = [
    ACC_NO_GROUP,
    ACC_SAME,
    ACC_CHOOSE_UPPER,
    ACC_CHOOSE_LOWER,
]