| 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) |