ESG-BERT-Reddit / configuration_ESGBertReddit.py
admation's picture
Update configuration_ESGBertReddit.py
8aba8d7
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)