File size: 432 Bytes
5fc6e5d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
"""
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",
]