File size: 380 Bytes
25c44d8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
from transformers import PretrainedConfig
class ESGRedditConfig(PretrainedConfig):
model_type = "ESGBertReddit"
def __init__(
self,
architectures = ["BertForSequenceClassification"],
num_classes: int = 4,
**kwargs
):
self.architectures = architectures
self.num_classes = num_classes
super().__init__(**kwargs) |