papri-ka's picture
Deploy FastAPI ML service to Hugging Face Spaces
5fc6e5d
raw
history blame contribute delete
432 Bytes
"""
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",
]