turing/modeling/models/__init__.py

"""
Model classes for code comment classification.
"""

from turing.modeling.models.codeBerta import CodeBERTa
from turing.modeling.models.graphCodeBert import GraphCodeBERTClassifier
from turing.modeling.models.randomForestTfIdf import RandomForestTfIdf
from turing.modeling.models.tinyBert import TinyBERTClassifier

__all__ = [
    "CodeBERTa",
    "RandomForestTfIdf",
    "TinyBERTClassifier",
    "GraphCodeBERTClassifier",
]