| # from .modeling_utils import ProteinConfig # noqa: F401 | |
| # from .modeling_utils import ProteinModel # noqa: F401 | |
| # from .modeling_bert import ProteinBertModel # noqa: F401 | |
| # from .modeling_bert import ProteinBertForMaskedLM # noqa: F401 | |
| # from .modeling_bert import ProteinBertForValuePrediction # noqa: F401 | |
| # from .modeling_bert import ProteinBertForSequenceClassification # noqa: F401 | |
| # from .modeling_bert import ProteinBertForSequenceToSequenceClassification # noqa: F401 | |
| # # TODO: ProteinBertForContactPrediction | |
| # from .modeling_resnet import ProteinResNetModel # noqa: F401 | |
| # from .modeling_resnet import ProteinResNetForMaskedLM # noqa: F401 | |
| # from .modeling_resnet import ProteinResNetForValuePrediction # noqa: F401 | |
| # from .modeling_resnet import ProteinResNetForSequenceClassification # noqa: F401 | |
| # from .modeling_resnet import ProteinResNetForSequenceToSequenceClassification # noqa: F401 | |
| # # TODO: ProteinResNetForContactPrediction | |
| # # TODO: ProteinLSTM* | |
| # from .modeling_unirep import UniRepModel # noqa: F401 | |
| # from .modeling_unirep import UniRepForLM # noqa: F401 | |
| # from .modeling_unirep import UniRepForValuePrediction # noqa: F401 | |
| # from .modeling_unirep import UniRepForSequenceClassification # noqa: F401 | |
| # from .modeling_unirep import UniRepForSequenceToSequenceClassification # noqa: F401 | |
| # # TODO: UniRepForContactPrediction | |
| # # TODO: Bepler* | |
| # from .modeling_onehot import OneHotModel # noqa: F401 | |
| # from .modeling_onehot import OneHotForValuePrediction # noqa: F401 | |
| # from .modeling_onehot import OneHotForSequenceClassification # noqa: F401 | |
| # from .modeling_onehot import OneHotForSequenceToSequenceClassification # noqa: F401 | |
| # TODO: OneHotForContactPrediction | |