Biopesticide-AI / bioai /models /__init__.py
flvcko's picture
Biopesticide-AI: AMD Hackathon Unicorn Track submission
914512c
Raw
History Blame Contribute Delete
468 Bytes
"""bioai.models -- neural network modules for the biopesticide-AI pipeline.
Re-exports the four public model classes so callers can simply do::
from bioai.models import SiRNACNN, CaduceusAdapter, DiscreteVAE, DegradationPINN
"""
from .sirna_cnn import SiRNACNN
from .caduceus_adapter import CaduceusAdapter
from .discrete_vae import DiscreteVAE
from .pinn_fate import DegradationPINN
__all__ = ["SiRNACNN", "CaduceusAdapter", "DiscreteVAE", "DegradationPINN"]