Skip to content

Models

Jump to bottom
Venkatesh Murugadas edited this page Mar 28, 2024 · 1 revision

python-Chebai employs deep neural network models for the semantic classification of chemical entities. Model classes are located within the chebai/models/ directory.

  • chebai/models/base.py: Contains the base class ChebaiBaseNet, inherited from the PyTorch Lightning module, facilitating custom model creation.

  • Example: chebai/models/electra.py showcases a custom model inherited from ChebaiBaseNet, implementing an Electra model.

Clone this wiki locally