from .attention_gru import AttentionGru
from .bilstm_attention import BilstmAttention
from .cnn import CnnModel1D, CnnModel2D, CnnModelRawData
from .cnn_features import CnnFeatures
from .crnn import CrnnModel
from .logistic_regression import LogisticRegression
from .lstm_attention import LstmAttention
from .svm import SvmModel

# MODEL NAME
CNN_MODEL_1D = 'cnn_1d'
CNN_MODEL_2D = 'cnn_2d'
CNN_MODEL_RAW_DATA = 'cnn_raw'
CNN_FEATURES = 'cnn_features'
CRNN_MODEL = 'crnn'
CRNN2D_MODEL = 'crnn2d'
CRNN2D_LARGER_MODEL = 'crnn2d_larger'
CRNN2D_VGG_MODEL = 'crnn2d_vgg'
SVM_MODEL = 'svm'
BILSTM_MODEL = 'bilstm'
LSTM_MODEL = 'lstm'
LR_MODEL = 'lr'
ATTGRU = 'att_gru'

from .attention_gru import AttentionGru
from .bilstm_attention import BilstmAttention
from .cnn import CnnModel1D, CnnModel2D, CnnModelRawData
from .cnn_features import CnnFeatures
from .crnn import CrnnModel
from .logistic_regression import LogisticRegression
from .lstm_attention import LstmAttention
from .svm import SvmModel

# MODEL NAME
CNN_MODEL_1D = 'cnn_1d'
CNN_MODEL_2D = 'cnn_2d'
CNN_MODEL_RAW_DATA = 'cnn_raw'
CNN_FEATURES = 'cnn_features'
CRNN_MODEL = 'crnn'
CRNN2D_MODEL = 'crnn2d'
CRNN2D_LARGER_MODEL = 'crnn2d_larger'
CRNN2D_VGG_MODEL = 'crnn2d_vgg'
SVM_MODEL = 'svm'
BILSTM_MODEL = 'bilstm'
LSTM_MODEL = 'lstm'
LR_MODEL = 'lr'
ATTGRU = 'att_gru'

# MODEL NAME