model/__init__.py

"""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"]