| from transformers import PretrainedConfig, AutoConfig | |
| from typing import List | |
| class MultiHeadCRFConfig(PretrainedConfig): | |
| model_type = "crf-tagger" | |
| def __init__( | |
| self, | |
| classes = list(), | |
| number_of_layer_per_head = 1, | |
| augmentation = "random", | |
| context_size = 64, | |
| percentage_tags = 0.2, | |
| aug_prob = 0.5, | |
| crf_reduction = "mean", | |
| freeze = False, | |
| version="0.1.3", | |
| **kwargs, | |
| ): | |
| self.classes = classes | |
| self.number_of_layer_per_head=number_of_layer_per_head | |
| self.version = version | |
| self.augmentation = augmentation | |
| self.context_size = context_size | |
| self.percentage_tags = percentage_tags | |
| self.aug_prob = aug_prob, | |
| self.crf_reduction = crf_reduction | |
| self.freeze=freeze | |
| self.version = version | |
| super().__init__(**kwargs) | |