ESG-BERT-Reddit / configuration_ESGBertReddit.py
admation's picture
Upload model
25c44d8
raw history blame
No virus
380 Bytes
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)