# -*- coding: utf-8 -*-
# file: __init__.py
# time: 02/11/2022 15:47
# author: YANG, HENG <hy345@exeter.ac.uk> (杨恒)
# github: https://github.com/yangheng95
# GScholar: https://scholar.google.com/citations?user=NPq5a_0AAAAJ&hl=en
# ResearchGate: https://www.researchgate.net/profile/Heng-Yang-17/research
# Copyright (C) 2022. All Rights Reserved.
[docs]
class GloVeProteinRModelList(list):
from .__classic__.cnn import CNN
from .__classic__.lstm import LSTM
from .__classic__.transformer import Transformer
from .__classic__.mhsa import MHSA
def __init__(self):
super(GloVeProteinRModelList, self).__init__(
[self.CNN, self.LSTM, self.Transformer, self.MHSA]
)
[docs]
class BERTProteinRModelList(list):
from .__plm__.bert import BERT_MLP
def __init__(self):
super(BERTProteinRModelList, self).__init__([self.BERT_MLP])