NABP-LSTM-Att / model /__init__.py
wuxing0105's picture
Upload folder using huggingface_hub
bf928ee verified
Raw
History Blame Contribute Delete
745 Bytes
"""NABP-LSTM-Att model package.
The compatibility aliases keep the class names stored in the upstream pickle
files (``features.InputFeatures``) importable after the source-tree cleanup.
"""
import sys
from . import tokenization as tokenization
sys.modules.setdefault("tokenization", tokenization)
from . import features as features
sys.modules.setdefault("features", features)
def get_model():
"""Build and compile the NABP-LSTM-Att network."""
from .network import get_model as build_model
return build_model()
def __getattr__(name):
if name == "AttLayer":
from .network import AttLayer
return AttLayer
raise AttributeError(name)
__all__ = ["AttLayer", "features", "get_model", "tokenization"]