NETWORKS = {1: "LSTM",
            2: "BiLSTMAttention",
            3: "GRU",
            4: "BOW",
            5: "CNN",
            6: "BiLSTM",
            7: "BiGRU",
            17: "ERNIE",
            }

LSTM = NETWORKS[1]
BiLSTMAttention = NETWORKS[2]
GRU = NETWORKS[3]
BOW = NETWORKS[4]
CNN = NETWORKS[5]
BiLSTM = NETWORKS[6]
BiGRU = NETWORKS[7]
# ERNIE = NETWORKS[7]
